diff --git a/deepspeed/utils/timer.py b/deepspeed/utils/timer.py index 19671add031f..9c98dc930d9e 100755 --- a/deepspeed/utils/timer.py +++ b/deepspeed/utils/timer.py @@ -88,6 +88,7 @@ def elapsed(self, reset=True): return elapsed_ def mean(self): + self.elapsed(reset=False) return trim_mean(self.elapsed_records, 0.1) def __init__(self):