pytorch模型保存和加载

84 主题	115 帖子	731 金钱

Rank: 9 Rank: 9 Rank: 9

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

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

GMT+8, 2024-11-22 23:03 , Processed in 0.197958 second(s), 22 queries .

		自动登录	找回密码
密码			立即注册