From 2c0c85d3d337c43748ddd3ea23dfd2417de79f11 Mon Sep 17 00:00:00 2001 From: Jiarui Fang Date: Wed, 5 Jan 2022 16:07:06 +0800 Subject: [PATCH] fix a bug in timer (#114) --- colossalai/trainer/_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/colossalai/trainer/_trainer.py b/colossalai/trainer/_trainer.py index e3257a20ee08..156d06862967 100644 --- 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,