Decimal rounding - logged metrics #19496
Unanswered
Arderiu
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Dear community,
In the training_step function of my LightningModule, I am logging the train and validation loss:
self.log('train_loss', loss, on_step=False, on_epoch=True, prog_bar=True)
However, the logged metrics are rounded to the nearest integer by default. Do you know how can I control the number of decimals shown? Thank you in advance!
Beta Was this translation helpful? Give feedback.
All reactions