From a407eb148c97dec3c8f8a3034e4ca7e7ee12de01 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Tue, 23 Feb 2021 09:17:08 +0000 Subject: [PATCH 1/2] put back --- android/test_app/make_assets.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/android/test_app/make_assets.py b/android/test_app/make_assets.py index ce5059b46ee..acec170753a 100644 --- a/android/test_app/make_assets.py +++ b/android/test_app/make_assets.py @@ -1,5 +1,6 @@ import torch import torchvision +from torch.utils.mobile_optimizer import optimize_for_mobile print(torch.__version__) @@ -12,6 +13,5 @@ model.eval() script_model = torch.jit.script(model) -# 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") +opt_script_model = optimize_for_mobile(script_model) +opt_script_model.save("app/src/main/assets/frcnn_mnetv3.pt") From 7095915811cb30d4fa51abc098e32decf49ae67c Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Tue, 23 Feb 2021 09:24:18 +0000 Subject: [PATCH 2/2] flake8 --- android/test_app/make_assets.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/android/test_app/make_assets.py b/android/test_app/make_assets.py index acec170753a..7860c759a57 100644 --- a/android/test_app/make_assets.py +++ b/android/test_app/make_assets.py @@ -1,6 +1,6 @@ import torch import torchvision -from torch.utils.mobile_optimizer import optimize_for_mobile +from torch.utils.mobile_optimizer import optimize_for_mobile print(torch.__version__) @@ -14,4 +14,4 @@ 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") +opt_script_model.save("app/src/main/assets/frcnn_mnetv3.pt")