Convert caffe model with PReLU layer

Machine Learning & AI Core ML
alex-golunov OP
Created Jul ’17
Replies 2
Boosts 0
Views 804
Participants 3

I've been trying to convert caffe model with PReLU activation function and getting the following error:

RuntimeError: Caffe model error in layer 'relu1a' of type 'PReLU': Expected a scalar parameter (alpha) when 'channel_shared' flag is set.

channel_shared is set to false.

Replies  2
Boosts  0
Views  804
Participants  3
mikhail_p OP
Jul ’17

Same error, this made me change PReLU on ReLU.

0
OP
Apple
Jul ’17

Hi Alex, Mikhail,


Sorry to hear that you are having issues with the PReLU layer. We have identified the issue and fixed it. The fix will be a part of the next coremltools package release (after 0.4.0).

Thank you for the patience.

0
Convert caffe model with PReLU layer
First post date Last post date
 
 
Q