如何在 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])