我的问题可以分为两部分。
是否有可以通过不同库使用的机器学习模型文件格式?比如我用pytorch保存了一个模型,然后用tensorflow加载?
如果没有,是否有可以帮助传输格式的库,以便可以在 keras 中直接使用 pytorch 机器学习模型?
之所以问这个问题,是因为最近需要将之前在tensorflow中训练的一些模型调整为pytorch。
这个问题的更新: Facebook 和微软将推出一个名为 ONNX 的模型标准,用于在不同框架之间传输模型,例如在 Pytorch 到 Caffe2 之间。链接如下:
这个问题的进一步更新: Tensorflow 本身使用 Protocol Buffer 格式来存储模型文件,可以用于不同模型之间的传输。链接如下: