如何在保存神经网络的同时解决路径问题

时间:2019-03-27 12:12:14

标签: python-3.x

我想将模型保存在pythorch中。但是colaboratory(colab.research.google.com)要求我输入无效的PATH语法。

文件“”,第7行     },“ abc.corpus”)     ^ SyntaxError:语法无效

我如何将其删除。

我想将urdu pos标记器作为一个项目。我将LSTM与CRF一起使用时具有91%的准确度。 我从那里使用代码,但这给了我一个路径错误

“”“”“ torch.save({             'epoch':时代,             'model_state_dict':model.state_dict(),             'optimizer_state_dict':optimizer.state_dict(),             “损失”:损失,             ...             },“ new.corpus”) “”“”“

“”“”“ torch.save(the_model.state_dict(),“ new.corpus”)

“”“”“

我希望模型必须保存在文件中,但是文件存在路径错误

1 个答案:

答案 0 :(得分:0)

  • 只有一个语法错误,这是对象的save的第一个参数,因此必须有大括号。
  • 路径(不是语料库)的文件类型是.pt或.pth

torch.save({the_model.state_dict()},“ new.pt”)