From 9b376c381f85f642defebb86b9e3e1a90d186f77 Mon Sep 17 00:00:00 2001 From: mrq Date: Sat, 11 Mar 2023 18:14:32 +0000 Subject: [PATCH] brain worm --- src/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/utils.py b/src/utils.py index cc2d4eb..d549145 100755 --- a/src/utils.py +++ b/src/utils.py @@ -670,7 +670,7 @@ class TrainingState(): self.steps = int(self.info['steps']) if 'iteration_rate' in self.info: - it_rate = self.info['iteration_rate'] / self.batch_size # why + it_rate = self.info['iteration_rate'] * self.batch_size # why self.it_rate = f'{"{:.3f}".format(1/it_rate)}it/s' if 0 < it_rate and it_rate < 1 else f'{"{:.3f}".format(it_rate)}s/it' self.it_rates += it_rate