Pytorch：如何在 OrderedDict 中将 requires_grad 更改为 true

如何解决Pytorch：如何在 OrderedDict 中将 requires_grad 更改为 true

假设我有一个来自 torch.nn 的神经网络对象，默认情况下 requires_grad 的参数是 False。我想将其更改为 True。但以下幼稚的方法失败了：

From torch import nn
a = nn.Linear(1,1)
a.state_dict()[‘weight’].requires_grad = True
print(a.state_dict()[‘weight’].requires_grad)

结果是False。任何人都可以解释问题是什么以及如何解决它？谢谢！我的火炬版本是 1.7.1。

解决方法

默认情况下可训练 nn 个对象参数 将具有 requires_grad=True。 您可以通过执行以下操作来验证：

import torch.nn as nn

layer = nn.Linear(1,1)

for param in layer.parameters():
    print(param.requires_grad)

# or use
print(layer.weight.requires_grad)
print(layer.bias.requires_grad)

要更改 requires_grad 状态：

for param in layer.parameters():
    param.requires_grad = False # or True