Prelu replacement - "Weights for scale layer" doesn't exists
ramanathan831 opened this issue · 1 comments
ramanathan831 commented
Hi,
Was the model trained on relu+scale combination or it was trained on prelu and you just replace the prelu with equivalent operations only in the prototxt file.
I have a model which was trained on prelu, but replacing it with relu+scale combination in the prototxt gives me a "Weights for scale layer" doesn't exists on Tensorrt. Any idea how to solve the issue?
PKUZHOU commented
it was trained on prelu, in fact the model prototxt and weight are borrowed from the official repo of MTCNN. If you replace with relu+scale, you should also modify the model weight to match your prototxt