Thanks for the update! Same issue is reported at Using D2Go's mask_rcnn model in the demo app · Issue #121 · pytorch/android-demo-app · GitHub 6. We will let you know once we get a solution for it. CC: @jeffxtang @fmassa