如何解决如何在PyTorch中使用模型的第一层
我已经上传了某个模型
from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained(model)
我可以看到模型:
print(model.state_dict())
该模型包含很多层,我只想获取前50个层。请告诉我该怎么做。
解决方法
我认为这应该可以解决问题:
model = nn.Sequential(*list(model.classifier.children())[:50])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。