是否可以在 TensorFlow 上加载学习模型（.t7）？

Posted 2023-03-29

【中文标题】是否可以在 TensorFlow 上加载学习模型（.t7）？

【英文标题】：Is it possible to load learned model (.t7) on TensorFlow?

【发布时间】：2018-06-22 15:32:33

【问题描述】：

我已经学习了 CNN 模型（model.t7），我想在另一个基于 Python 的深度学习框架（例如 TensorFlow）上使用这个模型。

你能告诉我如何实现吗？

谢谢。

【问题讨论】：

是的，有可能。您可以将模型参数保存为 numpy(*.npy) 的格式，因为 tf 支持 numpy 您可以通过加载 npy 文件来加载训练过的参数。当然，由于它是通过第三方包实现的，因此需要一些时间。

【参考方案1】：

那里有一些脚本/库。

您可以在此处找到概述：https://github.com/ysh329/deep-learning-model-convertor

在您的情况下，您需要转换：

从 Torch 到 pyTorch https://github.com/clcarwin/convert_torch_to_pytorch 从 pyTorch 到 Tensorflow（通过 Keras）https://github.com/nerox8664/pytorch2keras

但请事先阅读限制，因为并非所有操作都受支持。