pytorch
91b7bcf4 - [PyTorch Edge] Add proper error message when loading incompatible model with lite interpreter (#59354)

Commit
4 years ago
[PyTorch Edge] Add proper error message when loading incompatible model with lite interpreter (#59354) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59354 Check if the model has `bytecode.pkl` and provide proper error message before loading model. Test it by loading a model.pt and model.ptl. ``` >>> from torch.jit.mobile import _load_for_lite_interpreter >>> _load_for_lite_interpreter("/Users/chenlai/Documents/pytorch/data/mobilenet_v2.pt") Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/Users/chenlai/pytorch/torch/jit/mobile/__init__.py", line 48, in _load_for_lite_interpreter cpp_module = torch._C._load_for_lite_interpreter(f, map_location) # type: ignore[attr-defined] RuntimeError: The model is not generated from the api _save_for_lite_interpreter. Please regenerate the module by scripted_module._save_for_lite_interpreter('model.ptl'). Refer to https://pytorch.org/tutorials/prototype/lite_interpreter.html for more details. ``` iOS: ![image](https://user-images.githubusercontent.com/16430979/120593077-cbe23180-c3f3-11eb-9745-ee2b04b78c6c.png) Android: ![image](https://user-images.githubusercontent.com/16430979/120594357-af46f900-c3f5-11eb-9fb0-500a038148e3.png) Differential Revision: D28856713 D28856713 Test Plan: Imported from OSS Reviewed By: dhruvbird Pulled By: cccclai fbshipit-source-id: c3f9a3b64459dda6811d296371c8a2eaf22f8b20
Author
Parents
Loading