1
0
Fork 0
master
mrq 2023-03-05 20:13:39 +07:00
parent b2e89d8da3
commit b82961ba8a
1 changed files with 3 additions and 3 deletions

@ -760,9 +760,9 @@ class TrainingState():
self.metrics['step'] = [f"{self.epoch}/{self.epochs}"]
if self.epochs != self.its:
self.metric.append(f"{self.it}/{self.its}")
self.metrics.append(f"{self.it}/{self.its}")
if steps > 1:
self.metric.append(f"{step}/{steps}")
self.metrics.append(f"{step}/{steps}")
self.metrics['step'] = ", ".join(self.metrics['step'])
if lapsed: