Skip to content
Discussion options

You must be logged in to vote

You need to log it:

self.log("accuracy", accu, prog_bar=True)
return {'loss':loss}

Docs: https://pytorch-lightning.readthedocs.io/en/latest/extensions/logging.html#logging-from-a-lightningmodule

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by awaelchli
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment