caffepytorchpycaffe

PRelue is not supperted with mmdnn?

I converted my caffe model to IR successfully, error happened when I tried convert IR to pytorch:

Pytorch Emitter has not supported operator [PRelu]

How should I deal with that please?

Solution

  • Yes MMdnn support supports LeakyRelu. Check the link below for pytorch_emitter.py implementation from MMdnn.

    pytorch_emitter.py

    If you check the implementation you will find all the supported operations and it doesn't include PRelu.