python - 深度学习: save and load a universal machine model through different libraries

标签 python machine-learning tensorflow pytorch

我的问题可以分为两部分。

  • 是否有一种可以通过不同库使用的机器学习模型文件格式?例如,我通过pytorch保存了一个模型,然后使用tensorflow加载它?

  • 如果没有,是否有一个库可以帮助转换格式,以便 pytorch 机器学习模型可以直接在 keras 中使用?

之所以问这个问题,是因为最近需要将之前在tensorflow中训练的一些模型调整到pytorch。

此问题的更新: Facebook 和微软将推出一个名为 ONNX 的模型标准,用于在不同框架之间传输模型,例如 Pytorch 到 Caffe2 之间。链接如下:

https://research.fb.com/facebook-and-microsoft-introduce-new-open-ecosystem-for-interchangeable-ai-frameworks/

此问题的进一步更新: Tensorflow本身使用Protocol Buffer格式来存储模型文件,可以用于不同模型之间的传输。链接如下:

https://www.tensorflow.org/extend/tool_developers/

最佳答案

非常有趣的问题。神经网络是一种数学抽象,由层网络(卷积、循环等)、运算(点积、非线性等)及其各自的参数(权重、偏差)组成。

  1. 据我所知,没有通用模型文件。尽管如此,不同的库允许用户以二进制格式保存模型。

  2. 没有用于转换的库,但在 github repo 上有所努力。这解决了这个问题。

关于python - 深度学习: save and load a universal machine model through different libraries,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44318616/

相关文章:

r - 具有附加维度(子类)的二元分类

Python 将整数转换为日期

python - 在python3.4中加载.npy文件错误

python - 在 Python 中使用字符串格式化

python - Pytorch CNN错误: Expected input batch_size (4) to match target batch_size (64)

machine-learning - 在强化学习中设置 gamma 和 lambda

r - 使用 naiveBayes (e1071) 进行分类不起作用($levels 返回 NULL)

python - 如何将 tf.example 发送到 TensorFlow Serving gRPC 预测请求中

python - 用于匹配字母、数字和某些特定字符的正则表达式?

python - 如何使用 R 或 python 读取和转换 GrADS 平面二进制格式?