全文转载自:
https://discuss.pytorch.org/t/clearing-gradients-on-tensors-that-arent-a-part-of-a-model/50060/1
很简单,如果你看哪个tensor的梯度不顺眼,直接用None赋值就好
x.grad = None
这就是 torch 中的tensor的手动清除操作
全文转载自:
https://discuss.pytorch.org/t/clearing-gradients-on-tensors-that-arent-a-part-of-a-model/50060/1
很简单,如果你看哪个tensor的梯度不顺眼,直接用None赋值就好
x.grad = None
这就是 torch 中的tensor的手动清除操作