Solved by pchandrasekaran in post #13 
              
              
                @bibekx That’s a bit weird. I can’t tell how a variable change may have caused that. 
What I would suggest is: 

Check the model weights in the test_model function to ensure that an updated model is in fact passed.
Check that the dataloader in test_model is iterating through correctly.
Play around w…