pytorch保存模型和导入模型,torch.load出错是什么原因?

Song615 次浏览0个评论2018年07月11日
描述

Torch使用save都能成功,但是load不能成功是什么原因? [attach]27[/attach] [b]接下来我们看一下pytorch保存模型和导入模型[/b]   [code]# 保存和加载整个模型 torch.save(model_object, 'model.pkl') model = torch.load('model.pkl')[/code][code]# 仅保存和加载模型参数(推荐使用) torch.save(model_object.state_dict(), 'params.pkl') model_object.load_state_dict(torch.load('params.pkl'))[/code]

 

提交评论

请登录后评论

用户评论

    当前暂无评价,快来发表您的观点吧...

更多相关好文

    当前暂无更多相关好文推荐...

    公告提示

    Pytorch是Facebook的AI研究团队发布了一个Python工具包,是Python优先的深度学习框架。作为numpy的替代品;使用强大的GPU能力,提供最大的灵活性和速度,实现了机器学习框架Torch在Python语言环境的执行,基于python且具备强大GPU加速的张量和动态神经网络。