5

我想将 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 中加载 Torch 模型,但我得到了类似的错误。

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

谢谢

4

0 回答 0