github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-05 02:51:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix a bug in timer (#114)

Browse Source
This commit is contained in:
Jiarui Fang
2022-01-05 16:07:06 +08:00
committed by GitHub
parent 7904baf6e1
commit 2c0c85d3d3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/trainer/_trainer.py
View File

@@ -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,