Check if model is eval or train

2

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

73 Likes