代码之家 › 专栏 › 技术社区 › user836026

在pytorch中复制权重时语法无效

pytorch deep-learning keras tensorflow python

user836026 · 技术社区 · 3 年前

我试过这个:

model2.down1.maxpool_conv.1.double_conv.0.weight.copy_(state_dict['layer1.0.conv1.weight'])

但我得到了下面的错误信息:

model2.down1.maxpool_conv.1.double_conv.0.weight.copy_(state_dict['layer1.0.conv1.weight'])
                             ^
SyntaxError: invalid syntax

但我的模型输出显示它有“ 下降1。maxpool_conv.1。双转换。重量 "

1 回复 | 直到 3 年前

lejlot 3 年前

Python不允许数字因此,如果您以某种方式创建了一个具有此类属性的对象(只能通过低级函数调用来实现),您将无法通过点表示法访问这些属性(这就是为什么会出现无效语法错误,而不是类似于Attribute not found error的错误)。

你应该选择 param 引用循环中已有的内容,并进行复制

for name, param in model2.named_parameters():
  if name == 'down1.maxpool_conv.1.double_conv.0.weight':
     param.copy_(state_dict['layer1.0.conv1.weight'])

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

2 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

2 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

2 年前

Chris Ranck · Keras时间序列生成器-输入为“X”长,输出为“X-长”

2 年前

DPM · 将自定义掩码传递给LSTM数据以进行培训和验证

2 年前

Luciana Carota · 为什么Keras的R“预测”函数不能识别正确的输入维度?

2 年前

Sarah Elnaggar · 我使用Gensim Doc2vec进行图形嵌入,然后在keras中使用两层深度神经网络进行二元分类

2 年前

user836026 · 在pytorch中复制权重时语法无效

3 年前

mad · 如何从一个定制的神经网络模型中获得逻辑和概率

3 年前

Archit · 无法使用自定义距离函数加载带有lambda函数的keras模型

6 年前