fix a bug in timer (#114)

2c0c85d3 · Jiarui Fang · GitHub · 7904baf6 · 2c0c85d3
Unverified Commit 2c0c85d3 authored Jan 05, 2022 by Jiarui Fang Committed by GitHub Jan 05, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

colossalai/trainer/_trainer.py colossalai/trainer/_trainer.py +1 -1

No files found.
--- a/colossalai/trainer/_trainer.py
+++ b/colossalai/trainer/_trainer.py
@@ -193,7 +193,7 @@ class Trainer:

        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,