From 206a14fdbe978afd46aa3f0f8414b6a7c5466cbf Mon Sep 17 00:00:00 2001
From: mrq <barry.quiggles@protonmail.com>
Date: Sun, 5 Mar 2023 20:30:27 +0000
Subject: [PATCH] brianworms

---
 src/utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/utils.py b/src/utils.py
index 1633a76..9b3e5a0 100755
--- a/src/utils.py
+++ b/src/utils.py
@@ -760,9 +760,9 @@ class TrainingState():
 
 					self.metrics['step'] = [f"{self.epoch}/{self.epochs}"]
 					if self.epochs != self.its:
-						self.metrics.append(f"{self.it}/{self.its}")
+						self.metrics['step'].append(f"{self.it}/{self.its}")
 					if steps > 1:
-						self.metrics.append(f"{step}/{steps}")
+						self.metrics['step'].append(f"{step}/{steps}")
 					self.metrics['step'] = ", ".join(self.metrics['step'])
 
 			if lapsed:
@@ -808,7 +808,7 @@ class TrainingState():
 			self.metrics['loss'] = []
 
 			if 'learning_rate_gpt_0' in self.info:
-				self.metrics['loss'].append(f'LR: {"{:e}".format(self.info["learning_rate_gpt_0"])}')
+				self.metrics['loss'].append(f'LR: {"{:.3e}".format(self.info["learning_rate_gpt_0"])}')
 
 			if len(self.losses) > 0:
 				self.metrics['loss'].append(f'Loss: {"{:.3f}".format(self.losses[-1]["value"])}')