PyTorch Forums
F.leaky_relu blows up GPU
vision
peterjc123
(Pu Jiachen)
April 27, 2018, 2:19pm
4
Are you using 0.3.1? Please update to 0.4.0 because the memory leak has been solved.
Memory usage with pytorch-cpu in Windows
show post in topic