Hy, can you properly indent the code.
Both the functions are evaluating the model.



 Ka_Hina:

model.train()


This shouldn’t help, since the function is not training the model.



 Ka_Hina:

with torch.no_grad():


The second one is more better implementation