Thank you. Still I have problems on accuracy on validation set. I got around 25% at each epoch while the training accuracy increases. I set Shuffle = False on val test and Shuffle = True on train set. This is something wrong in how I call model.eval() ?