PyTorch Forums
Out of memory when optimizer.zero_grad() is called
david-leon
(David Leon)
March 16, 2017, 2:25am
6
The track back is the same, though.
What's the best way to handle exception "cuda runtime error (2) : out of memory"?
show post in topic