在PyTorch中手动更新神经网络的参数

时间:2018-09-19 17:31:31

标签: python neural-network pytorch

比方说,我有一个神经网络N,在训练它之后,我想手动更新其参数。例如,我想将其N.parameters()的元素的绝对值小于某个阈值归零。我怎样才能做到这一点? 如果我这样做

N.parameters()

它为我提供了一个生成器,我无法对其元素进行迭代。有没有更简单的方法来访问PyTorch中的参数?

0 个答案:

没有答案