diff --git a/android/test_app/make_assets.py b/android/test_app/make_assets.py index 7860c759a57..ce5059b46ee 100644 --- a/android/test_app/make_assets.py +++ b/android/test_app/make_assets.py @@ -1,6 +1,5 @@ import torch import torchvision -from torch.utils.mobile_optimizer import optimize_for_mobile print(torch.__version__) @@ -13,5 +12,6 @@ model.eval() script_model = torch.jit.script(model) -opt_script_model = optimize_for_mobile(script_model) -opt_script_model.save("app/src/main/assets/frcnn_mnetv3.pt") +# TODO: put back call to optimize_for_mobile once +# https://github.com/pytorch/pytorch/issues/52463 is fixed +script_model.save("app/src/main/assets/frcnn_mnetv3.pt")