Hello Mat's Archiver
论坛
›
深度学习
› pytorch模型保存和加载
GTX_AI
发表于 2020-12-25 21:27:36
pytorch模型保存和加载
.pt模型加载
net = torch.jit.load(model_path).to(device)
.pth模型加载
net.load_state_dict(torch.load(model_path))
页:
[1]
查看完整版本:
pytorch模型保存和加载