Pytorch, Pytorch Rnn, Pytorch Mnist Dataset, Build Model With Pytorch.
on test data We must call model.eval() to set dropout and batch normalization layers to evaluation mode before running inference. model.train() tells your model that you are training the model. So effectively layers like dropout, batchnorm etc. which behave different on the train and test