print in e-notation for LR
This commit is contained in:
parent
8b9c9e1bbf
commit
8094401a6d
|
@ -808,7 +808,7 @@ class TrainingState():
|
||||||
self.metrics['loss'] = []
|
self.metrics['loss'] = []
|
||||||
|
|
||||||
if 'learning_rate_gpt_0' in self.info:
|
if 'learning_rate_gpt_0' in self.info:
|
||||||
self.metrics['loss'].append(f'LR: {"{:.9f}".format(self.info["learning_rate_gpt_0"])}')
|
self.metrics['loss'].append(f'LR: {"{:e}".format(self.info["learning_rate_gpt_0"])}')
|
||||||
|
|
||||||
if len(self.losses) > 0:
|
if len(self.losses) > 0:
|
||||||
self.metrics['loss'].append(f'Loss: {"{:.3f}".format(self.losses[-1]["value"])}')
|
self.metrics['loss'].append(f'Loss: {"{:.3f}".format(self.losses[-1]["value"])}')
|
||||||
|
|
|
@ -1 +1 @@
|
||||||
Subproject commit 0f3261e0717be9e3a59b652fea7c1a388c8fa43a
|
Subproject commit 6fcd8c604f066e4e346da522bd14e6670395025f
|
Loading…
Reference in New Issue
Block a user