You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I am training the model using gpu, with batch size = 16, and everything is fine.
The cpu memory keeps growing during evaluation. How to fix it?
I've tried torch.no_grad() and torch.cuda.empty_cache() , but they don't work.
The text was updated successfully, but these errors were encountered:
I am training the model using gpu, with batch size = 16, and everything is fine.
The cpu memory keeps growing during evaluation. How to fix it?
I've tried
torch.no_grad()
andtorch.cuda.empty_cache()
, but they don't work.The text was updated successfully, but these errors were encountered: