深度学习良好的代码习惯

使用

with torch.enable_grad():

而不是

with torch.no_grad():

可以有效避免代码问题导致的梯度泄露。