mmdnn 不支持 PRelude？

Question

我成功地将我的caffe模型转换为IR，当我尝试将IR转换为pytorch时出现错误：

Pytorch Emitter has not supported operator [PRelu]

请问我该如何处理？

Answer 1

是的 MMdnn 支持支持 LeakyRelu。检查下面的 link 以了解来自 MMdnn 的 pytorch_emitter.py 实施。

pytorch_emitter.py

如果您检查实现，您会发现所有支持的操作，但不包括 PRelu。