weird epoch messaging
@@ -269,20 +269,13 @@ class TrainLoop():
exs_per_ep = self.world.num_examples()
if exs_per_ep:
- logs.append('total_epochs:{}'.format(
+ logs.append('epochs:{}'.format(
round(total_exs / exs_per_ep, 2)))
- exs_per_ep = self.world.num_examples()
- if exs_per_ep:
- logs.append('total_eps:{}'.format(
- round(self.world.get_total_exs() / exs_per_ep, 2)))
-
if 'time_left' in train_report:
logs.append('time_left:{}s'.format(
math.floor(train_report.pop('time_left', ""))))
- if 'num_epochs' in train_report:
- logs.append('num_epochs:{}'.format(
- train_report.pop('num_epochs', '')))
+
log = '[ {} ] {}'.format(' '.join(logs), train_report)
print(log)
self.log_time.reset()
