如何在PyTorch中使用模型的第一层

Question

我已经上传了某个模型

from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained(model)

我可以看到模型：

print(model.state_dict())

该模型包含很多层，我只想获取前50个层。请告诉我该怎么做。

Answer 1

我认为这应该可以解决问题：

model = nn.Sequential(*list(model.classifier.children())[:50])