Failed to Load Custom Model in Android

[pytorch]

Failed to Load Custom Model in Android

[dreiss]

We had some issues with snapshot uploads that should be resolved now. I just downloaded this repo, made these changes, and was able to run ./gradlew assembleDebug in the PyTorchDemoApp directory. If you're still experiencing this, please let us know. And please try to reproduce by checking out a clean copy of the repository and making these changes locally before building, and make sure that you're in the right directory (PyTorchDemoApp vs HelloWorldApp).

diff --git i/PyTorchDemoApp/app/build.gradle w/PyTorchDemoApp/app/build.gradle
index c0226ed..8ca451e 100644
--- i/PyTorchDemoApp/app/build.gradle
+++ w/PyTorchDemoApp/app/build.gradle
@@ -2,6 +2,9 @@ apply plugin: 'com.android.application'
 
 repositories {
     jcenter()
+    maven {
+        url "https://oss.sonatype.org/content/repositories/snapshots"
+    }
 }
 
 android {
@@ -35,6 +38,6 @@ dependencies {
     implementation "androidx.camera:camera-camera2:$camerax_version"
     implementation 'com.google.android.material:material:1.0.0-beta01'
 
-    implementation 'org.pytorch:pytorch_android:1.3.0'
-    implementation 'org.pytorch:pytorch_android_torchvision:1.3.0'
+    implementation 'org.pytorch:pytorch_android:1.4.0-SNAPSHOT'
+    implementation 'org.pytorch:pytorch_android_torchvision:1.4.0-SNAPSHOT'
 }

[ljk53]

@junjuew my fixes were landed and available in 1.4-snapshot build. You can update your build.gradle to try 1.4-snapshot, like this: ljk53@443b524

Please let me know if it fixes your problem, thanks!

cc: @ZhangWen0301 @hmzjwhmzjw

[junjuew]

@ljk53 I reproduced this error using an open-sourced neural style transfer network from here. The original repo is a bit convoluted, so I created a simple repo here to demonstrate the problem. The DNN definition is here. The README.md shows the steps of tracing in python and loading successfully in C++. However, when using the saved model in the android app, the error mentioned above occurs.

[IsaacBerman]

@ljk53 When I switch my dependencies to those found in the changes you linked "1.4.0-SNAPSHOT," i get the following errors "ERROR: Unable to resolve dependency for ':app@debug/compileClasspath': Failed to transform artifact 'pytorch_android.aar (org.pytorch:pytorch_android:1.4.0-SNAPSHOT:20191115.103440-37)' to match attributes {artifactType=jar}"

[ZhangWen0301]

same question

[hmzjwhmzjw]

same question

[ljk53]

@junjuew thanks for your feedback and it does look similar as #26764.
Are you able to repro this with some open source model? It would be helpful if I can read the model code and see what it does differently during model loading phase.

[ljk53]

@junjuew thanks for creating the easy repro steps. We are working on a fix now. I reopened #26764 and will keep updating there.

[junjuew]

@ljk53 It fixed my problem. Thank you. I'll go ahead and close this issue then.

[FilipJergleAlmquist]

I also get those errors when using 1.4.0-SNAPSHOT.

[Joyako]

@ljk53 The first few weeks, my project is OK when I used the "1.4.0-SNAPSHOT", but now it go wrong.The error is same to @IsaacBerman .

[Joyako]

@IsaacBerman You find any method to solve your question, if you have solved it , please tell me, thanks.

[IsaacBerman]

@Joyako No fix as of yet, will let you know if I find something.

[IsaacBerman]

@Joyako A project clean fixed the "Unable to resolve dependency error," but the original error still stands for me.