将PyNeural模型保存到磁盘
[英]Saving a PyNeural model to disk
问题描述
我使用PyNeural训练了模型。
如何保存这个训练有素的模型？ 我尝试了pickle和sklearn.externals.joblib但是没有用。 稍后，说TypeError: can't pickle NeuralNet objects
1 个解决方案
解决方案1
1 已采纳 2015-06-05 01:11:33
PyNeural的作者在这里。 最简单的方法是致电
NeuralNet.get_params()
您的NeuralNet对象上的方法。 这将返回一个由模型参数组成的元组。 您应该能够腌制和解开该元组。 要将参数加载到新模型中，请实例化具有相同大小和层数的新NeuralNet对象，然后调用
NeuralNet.set_params(params)
其中“ params”是您保存的元组。
希望您喜欢使用PyNeural！
pytorch：保存模型或state_dict提供了不同的磁盘空间占用
[英]Pytorch: saving model or state_dict gives different on-disk space occupation
声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.