昊月光华 2022-05-05 21:26 采纳率: 100%
浏览 292
已结题

pt模型转torchscript模型

在GitHub上找了个图像处理的工程,它自带了个pt文件,

model =a.LResNet50E_IR_Occ(num_mask=226)
test_model=torch.load("model_p5_w1_9938_9470_6503.pth.tar")
print(test_model.keys())

打印是
dict_keys(['state_dict', 'classifier'])
首先,初识模型转换,加载它自带的这个pt文件的model有state_dict和classifer两个键,第一个问题:
如果想转换成torchscript是不是只把state_dict加载进本工程的网络结构就可?state_dict对应的值又是一个 <class 'collections.OrderedDict'> ,键是各个网络层的权重。
第二个问题:
通用的将pt模型转torchScript的步骤是什么?

img


我的这个步骤有没有问题?

  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2022-05-06 09:47
    关注


    通常普通的简单步骤可以看官网的实例,
    一般来说你要先保证你的模型能够正常运行出结果,然后直接
    ts = torch.jit.trace(model, input_, strict=False)
    ts.save(path/to/torchscript)
    比较麻烦的是这个_extra_files,看你自己的网络需要存哪些东西下来了,如果不需要的话(也就是网络有些参数你去不知道,不需要模型帮你保存,就忽略)

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 5月15日
  • 已采纳回答 5月7日
  • 创建了问题 5月5日

悬赏问题

  • ¥15 数据库获取信息反馈出错,直接查询了ref字段并且还使用了User文档的_id而不是自己的
  • ¥15 将安全信息用到以下对象时发生以下错误:c:dumpstack.log.tmp 另一个程序正在使用此文件,因此无法访问
  • ¥15 速度位置规划实现精确定位的问题
  • ¥15 代码问题:df = pd.read_excel('c:\User\18343\Desktop\wpsdata.xlxs')路径读不到
  • ¥15 为什么视频算法现在全是动作识别?
  • ¥15 编写一段matlab代码
  • ¥15 用Python做岩石类别鉴定软件
  • ¥15 关于调取、提交更新数据库记录的问题
  • ¥15 之前删了盘从下vs2022遇见这个问题 搞了一整天了
  • ¥15 从Freecad中宏下载的DesignSPHysics,出现如下问题是什么原因导致的(语言-python)