pytorch加载官网预训练模型后print为参数

pytorch官网提供的只有参数

如果直接load模型print只有参数

import torch
import torchvision.models as models

# pretrained=True 使用预训练的模型
net = models.resnet101(pretrained=False)
pthfile = 'resnet101-5d3b4d8f.pth'
net.load_state_dict(torch.load(pthfile))
print(net)

print出来就是网络模型

发布了31 篇原创文章 · 获赞 5 · 访问量 5万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章