JIT android issue

I have created JIT model of the object detection SSD network and it runs fine when I run the traced model on Python or C++. I am following PyTorch mobile’s tutorial to load the network on phone. But when I do forward pass of the same traced model in Java environment of Android Studio, it is waits on that line for hours and the execution is completed. Anyone with any pointers what might be the issue?
Help appreciated.

Hello @vishalagarwal ,
Which version of pytorch android api do you use 1.3, 1.3.1 or nightly builds?

We had a deadlock issue that was fixed in https://github.com/pytorch/pytorch/pull/29885

If you are not using nightly builds, could you please check if it reproduces on nightlies?(https://github.com/pytorch/pytorch/tree/master/android#nightly)

1 Like

I am using pytorch 1.3.1
Let me try with nightly build

It resolved with the nightly build. Thanks!

1 Like