Thanks for your answer! What about the combination of the two commands (both ‘self.training options with/without gradient)?
I am a little bit confuse with the possibility of using back-propagation in the evaluation mode