github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix a bug in timer (#114)

pull/130/head
Jiarui Fang 2022-01-05 16:07:06 +08:00 committed by GitHub
parent 7904baf6e1
commit 2c0c85d3d3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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,