Skip to content

Commit 5de19de

Browse files
authored
Put back call to optimze_for_mobile (#3431)
* put back * flake8
1 parent 0e32a12 commit 5de19de

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)