Skip to content

Commit e1c49fa

Browse files
authored
Removed call to optimze_for_mobile (#3424)
* removed call to optimze_for_mobile * removed unrelated changes
1 parent 62e185c commit e1c49fa

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,6 +1,5 @@
11
import torch
22
import torchvision
3-
from torch.utils.mobile_optimizer import optimize_for_mobile
43

54
print(torch.__version__)
65

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

1413
model.eval()
1514
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")
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")

0 commit comments

Comments
 (0)