将nngraph模型转换为nn模型

时间:2017-06-14 07:12:42

标签: caffe torch pytorch caffe2

我想将pix2pix图像转换为使用nngraph构建的图像转换模型(https://github.com/phillipi/pix2pix)。

当我尝试使用torch to caffe工具(https://github.com/facebook/fb-caffe-exts#torch2caffe)将其转换为caffe模型时,我收到错误“unknown class nn.gModule”。

我还尝试通过torch.utils.serialization中的load_lua在pytorch中加载火炬模型,我得到类似的错误。

因为,转换似乎不支持nngraph。有没有办法将nngraph模型转换为nn模型而无需重写或重新训练?接下来,我想将pix2pix模型转换为各种格式,如caffe,caffe2,coreML等,以测试移动环境中的性能。

由于

0 个答案:

没有答案
相关问题