Skip to content

Commit fb8414b

Browse files
fmassafacebook-github-bot
authored andcommitted
Put back call to optimze_for_mobile (#3431)
Summary: * put back * flake8 Reviewed By: NicolasHug Differential Revision: D26605313 fbshipit-source-id: 6bf2177761b181e285320eaee2fc0d0fb1f092cc
1 parent 8a12399 commit fb8414b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

android/test_app/make_assets.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import torch
22
import torchvision
3+
from torch.utils.mobile_optimizer import optimize_for_mobile
34

45
print(torch.__version__)
56

@@ -12,6 +13,5 @@
1213

1314
model.eval()
1415
script_model = torch.jit.script(model)
15-
# 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")
16+
opt_script_model = optimize_for_mobile(script_model)
17+
opt_script_model.save("app/src/main/assets/frcnn_mnetv3.pt")

0 commit comments

Comments
 (0)