pytorch
diff --git a/‎HelloWorld/model.pt‎
-13.3 MB b/‎HelloWorld/model.pt‎
-13.3 MB
diff --git a/‎HelloWorld/trace_model.py‎
Lines changed: 1 addition & 1 deletion b/‎HelloWorld/trace_model.py‎
Lines changed: 1 addition & 1 deletion
@@ -7,4 +7,4 @@
 example = torch.rand(1, 3, 224, 224)
 traced_script_module = torch.jit.trace(model, example)
 torchscript_model_optimized = optimize_for_mobile(traced_script_module)
-torchscript_model_optimized.save("HelloWorld/HelloWorld/model/model.pt")
+torchscript_model_optimized._save_for_lite_interpreter("HelloWorld/HelloWorld/model/model.pt")