Trace ssd decode function

Hi everyone. I need to deploy ssd object detection model on android, but since torch ops does not available on android. I need to trace decode function ( to decode detection result. but I do not manage to trace it. Does any one succeed to trace the decode part ?

I tried to trace the function but I got the following error:

Compiled from code decode forward
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/ _slow_forward
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/ call
/usr/local/lib/python3.6/dist-packages/torch/jit/ trace_module
/usr/local/lib/python3.6/dist-packages/torch/jit/ trace

The above operation failed in interpreter, with the following stack trace:

We don’t have a recipe for custom ops yet. Stay turned!

1 Like