Sure @guangy10 , Acutally I’m using the demo app (provided by PyTorch) for Image segmentation
provided here 2 on GitHub, which is using deeplab v3, I replaced my u2net.ptl model with deeplab v3 model and its giving me errors like follows:

java.lang.IllegalStateException: Expected IValue type 13, actual type 7
at org.pytorch.IValue.preconditionType(IValue.java:314)
at org.pytorch.IValue.toDictStringKey(IValue.java:301)
at org.pytorch.imagesegmentation.MainActivity.run(MainActivity.java:124)
at java.lang.Thread.run(Thread.java:919)