完整代码
from torch import nn
conv2d = nn.Conv2d(3, 1, kernel_size=11, stride=4, padding=2)
weight = par['features.0.weight']
weight = weight[0]
conv2d.weight=torch.nn.Parameter(weight)# 需要先转为torch.nn.Parameter类型,直接赋值报错:cannot assign 'torch.FloatTensor' as parameter 'weight' (torch.nn.Parameter or None expected)
拓展:pytorch的参数_初始化
self.conv1 = nn.Conv2d(3, 64, kernel_size=7, stride=2, padding=3)
init.xavier_uniform(self.conv1.weight)
init.constant(self.conv1.bias, 0.1)
1万+

被折叠的 条评论
为什么被折叠?



