Out of GPU Memory during test, Model ouput (eval) & require_grad = True

2

Hi,

You have some more informations 'model.eval()' vs 'with torch.no_grad()') .
Let me know if you have more questions.

2 Likes