属性错误： 'Model'对象没有属性'parameters'

Question

我使用的是一个经过修改的Resnet18，在Resnet的末尾有我自己的池函数。

这是我的代码：

resnet = resnet18().cuda() #a modified resnet
class Model():
def __init__(self, model, pool):
self.model = model
self.pool= pool #my own pool class which has trainable layers
def forward(self, sample):
output = self.model(sample)
output = self.pool(output)
output = F.normalize(output, p=2, dim=1)
return output

现在，很明显，我不仅需要训练resnet部分，还需要训练pool部分。

但是，当我检查时：

model = Model(model=resnet, pool= pool)
print(list(model.parameters()))

它给出：

AttributeError: 'Model' object has no attribute 'parameters'

有人能帮忙吗？

Answer 1

您需要Model来继承torch.nn.Module:

class Model(torch.nn.Module):
def __init__(self, model, pool):
super(Model, self).__init__()
...