Check if model is eval or train

How can one check is a model is in train or eval state?

18 Likes

module.training is the boolean you are looking for. :slight_smile:

53 Likes

That’s easy! Thank you!

1 Like

great! thanks. just solve my issue.

1 Like

me too, this just solved my issue, thanks Simon!