我的问题可以分为两部分。
是否有一种可以通过不同库使用的机器学习模型文件格式?例如,我通过pytorch保存了一个模型,然后使用tensorflow加载它?
如果没有,是否有一个库可以帮助转换格式,以便 pytorch 机器学习模型可以直接在 keras 中使用?
之所以问这个问题,是因为最近需要将之前在tensorflow中训练的一些模型调整到pytorch。
此问题的更新: Facebook 和微软将推出一个名为 ONNX 的模型标准,用于在不同框架之间传输模型,例如 Pytorch 到 Caffe2 之间。链接如下:
此问题的进一步更新: Tensorflow本身使用Protocol Buffer格式来存储模型文件,可以用于不同模型之间的传输。链接如下:
最佳答案
非常有趣的问题。神经网络是一种数学抽象,由层网络(卷积、循环等)、运算(点积、非线性等)及其各自的参数(权重、偏差)组成。
据我所知,没有通用模型文件。尽管如此,不同的库允许用户以二进制格式保存模型。
没有用于转换的库,但在 github repo 上有所努力。这解决了这个问题。
关于python - 深度学习: save and load a universal machine model through different libraries,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44318616/