fix a bug in timer (hpcaitech#114)

erhaa233 · Jan 5, 2022 · 2c0c85d · 2c0c85d
1 parent 7904baf
commit 2c0c85d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colossalai/trainer/_trainer.py b/colossalai/trainer/_trainer.py
@@ -193,7 +193,7 @@ def _train_epoch(self,
 
         self._call_timer(action='stop', item='Train-epoch', keep_in_history=True)
         self._call_hooks('after_train_epoch')
-        self._call_timer(action='reset', item='Train-step')
+        self._call_timer(action='reset', item='Train-epoch')
 
     def _eval(self,
               test_dataloader: DataLoader,