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
-
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