How can one check is a model is in train or eval state?
21 Likes
module.training is the boolean you are looking for. 
73 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!