We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8a2dc6f commit 506279cCopy full SHA for 506279c
android/test_app/make_assets.py
@@ -1,5 +1,6 @@
1
import torch
2
import torchvision
3
+from torch.utils.mobile_optimizer import optimize_for_mobile
4
5
print(torch.__version__)
6
@@ -12,6 +13,5 @@
12
13
14
model.eval()
15
script_model = torch.jit.script(model)
-# TODO: put back call to optimize_for_mobile once
16
-# https://github.com/pytorch/pytorch/issues/52463 is fixed
17
-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")
0 commit comments