- 博客(1)
- 收藏
- 关注
RSS订阅原创 pytorch的pt模型保存与调用
pytorch的pt模型保存仅保存和加载模型参数# 模型保存checkpoint={'epoch':epoch, 'best_loss':best_loss, 'model':model.state_dict() 'optimizer':optimizer.state_dict()}torch.save(checkpoint, PATH='model.pth') #只保存模型权重参数，不保存模型结构# 模型调用the_model = TheModelClass(*ar
2020-11-11 11:11:31
19085
4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人