pytorch / android-demo-app

PyTorch android examples of usage in applications
1.46k stars 604 forks source link

Problem with '_save_for_lite_interpreter' function #223

Open ir1979 opened 2 years ago

ir1979 commented 2 years ago

In Image_segmentation task, the generated .ptl file by _save_for_lite_interpreter produces a model that outputs a [1,1000] tensor instead of a DICT_STRING_KEY. In the android source, if we replace the generated file with the file before optimization, the application works correctly.