update

qywu · Nov 12, 2021 · 562889a · 562889a
1 parent 90e4195
commit 562889a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/torchfly/training/trainer.py b/torchfly/training/trainer.py
@@ -56,10 +56,10 @@ def __init__(self, config: DictConfig, model: FlyModel, *args, **kwargs):
         self.total_num_batches = None
         self.total_num_epochs = None
         self.epoch_num_batches = None
-        self.global_batch_count = 0
-        self.global_step_count = 0
-        self.epochs_trained = 0
-        self.local_step_count = 0
+        self.global_batch_count = None
+        self.global_step_count = None
+        self.epochs_trained = None
+        self.local_step_count = None
 
         self.init_distributed_environment()