No, you should use it for calculating the accuracy of the validation set.

I got it. I’ll do this way.

Did you set it again after loading the model?

Could you tell me what “it” mean please?
When I loaded saved model, I did following way:
model = MyModel()
model.load_state_dict...
model.train(True)
...learning...
model.train(False)
...evaluating...