brain worm

2023-03-11 18:14:32 +07:00 · 2023-03-11 18:14:32 +07:00 · 9b376c381f
parent 94551fb9ac
commit 9b376c381f
1 changed files with 1 additions and 1 deletions
--- 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