@@ -231,7 +231,9 @@ def train():
                 avg_time = (time.time() - time_start) / (step + 1)
                 eta = str(datetime.timedelta(seconds=int(avg_time * (total_steps - global_step - 1))))
                 eta_epoch = str(datetime.timedelta(
-                    seconds=int(avg_time * (train_dataloader_len - sampler.step_start - step - 1))
+                    seconds=int(
+                        avg_time * (train_dataloader_len - sampler.step_start // config.train_batch_size - step - 1)
+                    )
                 ))
                 log_buffer.average()