Check if model is eval or train

1

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

2

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

3

That’s easy! Thank you!

4

great! thanks. just solve my issue.

5

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