小编Ант*_*тон的帖子

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

我已经上传了某个模型

from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained(model)
Run Code Online (Sandbox Code Playgroud)

我可以看到模型:

print(model.state_dict())
Run Code Online (Sandbox Code Playgroud)

该模型包含相当多的层,我只想获取前 50 层。请告诉我如何做到这一点。

python pytorch

1
推荐指数
1
解决办法
1450
查看次数

标签 统计

python ×1

pytorch ×1