caffe - 将 nngraph 模型转换为 nn 模型

Question

我想将 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 等，以测试在移动环境中的性能。

谢谢