代码之家 › 专栏 › 技术社区 › havakok

从任何平台加载和使用预先训练好的神经网络

pre-trained-model neural-network

1

havakok · 技术社区 · 7 年前

我正在构建一个代码,并尽可能地保持通用性。我看过很多教程和帖子,但它们都是特定于平台的(tensorflow\pytorch)。

有没有一个好的方法来加载和使用一个以前训练过的神经网络模型,使代码能够同时处理torch和tensorflow?网络构建在哪个版本的tensorflow\torch中有关系吗?我希望代码尽可能通用。

1 回复 | 直到 7 年前

1

2

user2997100 7 年前

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

您可能想研究一下开放式神经网络交换格式( https://onnx.ai/ )如果你是自己创建模型的话,这是一个由Amazon、Facebook、Microsoft和其他公司支持的为深度学习模型创建可移植文件格式的计划。