pytorch:inception v3参数为空错误
我正在使用torchvision.models中的inception_v3作为我的基础模型,并在最后添加一个FC层以获取功能。但是,我得到一个空参数错误pytorch:inception v3参数为空错误,pytorch,Pytorch,我正在使用torchvision.models中的inception_v3作为我的基础模型,并在最后添加一个FC层以获取功能。但是,我得到一个空参数错误 import torch import torch.nn as nn import torchvision.models as models class Baseline(nn.Module): def __init__(self, out_size): super().__init__() model
import torch
import torch.nn as nn
import torchvision.models as models
class Baseline(nn.Module):
def __init__(self, out_size):
super().__init__()
model = models.inception_v3(pretrained=True)
model.fc = nn.Linear(2048, out_size)
model.aux_logits = False
# Freeze model weights
for param in model.parameters():
param.requires_grad = False
self.parameters = nn.ParameterList()
def forward(self, image):
x = model(image)
x = x.view(x.size(0), -1)
x = model.fc = (x)
return x
我的理解是,您正在使用空的
nn.ParameterList
更新self.parameters
,此处不需要该列表
self.parameters
将已经拥有基线类拥有的所有参数,包括inception\u v3
和nn.Linear
的参数。当您在末尾用空列表更新它们时,实际上是在删除以前存储的所有参数