首页 > 解决方案 > 从任何平台加载和使用预训练的神经网络

问题描述

我正在构建代码并尝试使事情尽可能通用。我看过许多教程和帖子,但它们都是特定于平台的(tensorflow\pytorch)。

有没有一种好方法可以加载和使用先前训练的神经网络模型,使代码能够同时处理火炬和张量流?网络内置在哪个版本的 tensorflow\torch 中是否重要?我希望代码尽可能通用。

另外,我是否需要知道原始网络的结构,或者我可以加载它并在没有结构概念的情况下使用它?

标签: neural-networkpre-trained-model

解决方案


我认为不可能编写一个程序来从 Torch 和 Tensorflow 加载预先训练的模型,因为它们以不同的格式保存。

如果您自己创建模型,您可能需要查看开放神经网络交换格式 ( https://onnx.ai/ ),这是由 Amazon、Facebook、Microsoft 和其他公司支持的创建可移植文件格式的倡议用于深度学习模型。


推荐阅读