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 62e185c commit e1c49faCopy full SHA for e1c49fa
android/test_app/make_assets.py
@@ -1,6 +1,5 @@
1
import torch
2
import torchvision
3
-from torch.utils.mobile_optimizer import optimize_for_mobile
4
5
print(torch.__version__)
6
@@ -13,5 +12,6 @@
13
12
14
model.eval()
15
script_model = torch.jit.script(model)
16
-opt_script_model = optimize_for_mobile(script_model)
17
-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")
0 commit comments