[转载]torch中手动清除梯度

全文转载自:
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的手动清除操作

猜你喜欢

转载自blog.csdn.net/HaoZiHuang/article/details/129560472
今日推荐