从任何平台加载和使用预训练的神经网络
loading and using a pre-trained neural network from any platform
我正在构建代码并尽量保持通用性。我看过许多教程和 post,但它们都是特定于平台的 (tensorflow\pytorch)。
是否有一种好的方式来加载和使用以前训练过的神经网络模型,使代码能够同时处理 torch 和 tensorflow?网络内置于哪个 tensorflow\torch 版本中重要吗?我希望代码尽可能通用。
另外,我需要知道原始网络的结构吗?或者我可以在没有结构概念的情况下加载它并使用它吗?
我认为不可能编写可以从 Torch 和 Tensorflow 加载预训练模型的程序,因为它们以不同的格式保存。
如果您自己创建模型,您可能需要查看开放神经网络交换格式 (https://onnx.ai/),这是一项由 Amazon、Facebook、Microsoft 和其他公司支持的倡议,旨在创建一个可移植的深度学习模型的文件格式。
我正在构建代码并尽量保持通用性。我看过许多教程和 post,但它们都是特定于平台的 (tensorflow\pytorch)。
是否有一种好的方式来加载和使用以前训练过的神经网络模型,使代码能够同时处理 torch 和 tensorflow?网络内置于哪个 tensorflow\torch 版本中重要吗?我希望代码尽可能通用。
另外,我需要知道原始网络的结构吗?或者我可以在没有结构概念的情况下加载它并使用它吗?
我认为不可能编写可以从 Torch 和 Tensorflow 加载预训练模型的程序,因为它们以不同的格式保存。
如果您自己创建模型,您可能需要查看开放神经网络交换格式 (https://onnx.ai/),这是一项由 Amazon、Facebook、Microsoft 和其他公司支持的倡议,旨在创建一个可移植的深度学习模型的文件格式。