Solved by ptrblck in post #2 
              
              
                eval() changes the Module to evaluation, i.e. some layers like Dropout and BatchNorm change their behavior. In the case of Dropout, the connections won’t be dropped but scaled. 


The curves show the predictions on the test data. As you can see, the “overfitted” model predicts the samples closer t…