pytorch 加载模型和参数,不依赖源码和模model.py文件

torch.save(model,“模型名字”)

torch.load 是需要依赖源码的,pickle.load 调用 就会找模型定义的类,就会根据路径找,
所以需要在同一个 工程项目里面,否则报错, 还是需要工程文件 相关代码文件

不依赖 源码代码 只用一个模型在任意环境加载的办法 torchscript

c++环境中调用pytorch模型

trace 和 script

trace_module = torch.jit.trace(model,torch.rand(1,1,224,224)) 
print(trace_module.code)  # 查看模型结构
output = trace_module (torch.ones(1, 3, 224, 224)) # 测试
print(output)
trace_modult('model.pt') # 模型保存

# 此时应该用script方法  模型定义有 if else  等控制语句
script_module = torch.jit.script(model) 
print(script_module.code)
output = script_module(torch.rand(1,1,224,224))
print(output)
script_modult('model.pt') # 模型保存

pytorch模型转torchscript

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值