pytorch两种模型保存方式
只保存模型参数
# 保存 torch.save(model.state_dict(), '\parameter.pkl') # 加载 model = TheModelClass(...) model.load_state_dict(torch.load('\parameter.pkl'))
保存完整模型
# 保存 torch.save(model, '\model.pkl') # 加载 model = torch.load('\model.pkl')
多思考也是一种努力,做出正确的分析和选择,因为我们的时间和精力都有限,所以把时间花在更有价值的地方。