u014205537 2020-10-13 22:39 采纳率: 0%
浏览 294

pytorch如何加载自己训练的模型

在resnet上添加了注意力机制然后训练得到了一个.pth格式的模型,我想请问下,如何加载这个模型并打印这个模型的权重。
如,
import torchvision.models as models

model = models.vgg16(pretrained = True)

print(model)
这个可以输出vgg16的模型权重层。

  • 写回答

0条回答 默认 最新

    报告相同问题?

    悬赏问题

    • ¥100 set_link_state
    • ¥15 虚幻5 UE美术毛发渲染
    • ¥15 CVRP 图论 物流运输优化
    • ¥15 Tableau online 嵌入ppt失败
    • ¥100 支付宝网页转账系统不识别账号
    • ¥15 基于单片机的靶位控制系统
    • ¥15 真我手机蓝牙传输进度消息被关闭了,怎么打开?(关键词-消息通知)
    • ¥15 装 pytorch 的时候出了好多问题,遇到这种情况怎么处理?
    • ¥20 IOS游览器某宝手机网页版自动立即购买JavaScript脚本
    • ¥15 手机接入宽带网线,如何释放宽带全部速度