请选择 进入手机版 | 继续访问电脑版

Hello Mat

 找回密码
 立即注册
查看: 6731|回复: 0

pytorch模型保存和加载

[复制链接]

84

主题

115

帖子

731

金钱

管理员

Rank: 9Rank: 9Rank: 9

积分
1467
发表于 2020-12-25 21:27:36 | 显示全部楼层 |阅读模式

.pt模型加载
    net = torch.jit.load(model_path).to(device)
.pth模型加载
    net.load_state_dict(torch.load(model_path))


回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Python|Opencv|MATLAB|Halcom.cn ( 蜀ICP备16027072号 )

GMT+8, 2024-3-28 21:44 , Processed in 0.201811 second(s), 21 queries .

Powered by Discuz! X3.4

Copyright © 2001-2021, Tencent Cloud.

快速回复 返回顶部 返回列表