prevent division by zero (#3555)

facebookresearch · Mar 26, 2021 · 91cd630 · 91cd630
1 parent bbd6e26
commit 91cd630
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parlai/scripts/train_model.py b/parlai/scripts/train_model.py
@@ -681,7 +681,7 @@ def _compute_eta(
                 eta = time_left
 
         max_train_steps = self.opt.get('max_train_steps', -1)
-        if max_train_steps > 0:
+        if max_train_steps > 0 and steps_taken > 0:
             steps_progress = steps_taken / max_train_steps
             eta = (1 - steps_progress) * time_elapsed / steps_progress