Faced some issue while calling performInference for mobileBert on Android

[Nittur]

Abort message: 'terminating due to uncaught exception of type Ort::Exception: Non-zero status code returned while running Gather node. Name:'/model/mobilebert/embeddings/token_type_embeddings/Gather' Status Message: indices element out of data bounds, idx=98784247808 must be within the inclusive range [-2,1]'

Can someone help me understand what it means and where i might have gone wrong? I have rechecked the token input which im giving, it has worked in the PC side for the same onnx models.

Also is it possible to train mobilebert model on android? If anyone has tried this...please provide method used.

[edgchen1]

The error message is referring to a runtime check in a Gather node.

Name:'/model/mobilebert/embeddings/token_type_embeddings/Gather' Status Message: indices element out of data bounds, idx=98784247808 must be within the inclusive range [-2,1]'

The indices element value of 98784247808 is out of range. This value doesn't seem correct. You might try looking into where it is coming from.