Skip to content

GitLab

Unverified Commit e1c49faf authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Removed call to optimze_for_mobile (#3424) 

* removed call to optimze_for_mobile

* removed unrelated changes
parent 62e185c7
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
android/test_app/make_assets.py
View file @ e1c49faf
import torch import torch
import torchvision import torchvision
from torch.utils.mobile_optimizer import optimize_for_mobile
print(torch.__version__) print(torch.__version__)
...@@ -13,5 +12,6 @@ model = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_320_fpn( ...@@ -13,5 +12,6 @@ model = torchvision.models.detection.fasterrcnn_mobilenet_v3_large_320_fpn(
model.eval() model.eval()
script_model = torch.jit.script(model) script_model = torch.jit.script(model)
opt_script_model = optimize_for_mobile(script_model) # TODO: put back call to optimize_for_mobile once
opt_script_model.save("app/src/main/assets/frcnn_mnetv3.pt") # https://github.com/pytorch/pytorch/issues/52463 is fixed
script_model.save("app/src/main/assets/frcnn_mnetv3.pt")
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment