This commit is contained in:
mrq 2023-03-05 20:13:39 +00:00
parent b2e89d8da3
commit b82961ba8a

View File

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