Commit c04ef895 authored by Jacob Szwejbka's avatar Jacob Szwejbka Committed by Facebook GitHub Bot
Browse files

Remove methods_to_optimize from script

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/54

This arg is being deprecated, and its use case was really only for modules that use functions besides forward for inference.  The new plan is just to optimize every function. Since this script was just created I'm hoping I can edit this without throwing lots of stuff out of wack.

Reviewed By: wat3rBro

Differential Revision: D27954176

fbshipit-source-id: fbe178fcc0404e5d2524d8edb4052e2cd17f43ba
parent 42d34842
...@@ -24,7 +24,6 @@ class MobileOptimizationConfig(NamedTuple): ...@@ -24,7 +24,6 @@ class MobileOptimizationConfig(NamedTuple):
optimization_blocklist: Set[MobileOptimizerType] = None optimization_blocklist: Set[MobileOptimizerType] = None
preserved_methods: List[AnyStr] = None preserved_methods: List[AnyStr] = None
backend: str = "CPU" backend: str = "CPU"
methods_to_optimize: List[AnyStr] = None
def trace_and_save_torchscript( def trace_and_save_torchscript(
...@@ -68,7 +67,6 @@ def trace_and_save_torchscript( ...@@ -68,7 +67,6 @@ def trace_and_save_torchscript(
optimization_blocklist=mobile_optimization.optimization_blocklist, optimization_blocklist=mobile_optimization.optimization_blocklist,
preserved_methods=mobile_optimization.preserved_methods, preserved_methods=mobile_optimization.preserved_methods,
backend=mobile_optimization.backend, backend=mobile_optimization.backend,
methods_to_optimize=mobile_optimization.methods_to_optimize,
) )
with _synced_local_file("mobile_optimized.ptl") as lite_path: with _synced_local_file("mobile_optimized.ptl") as lite_path:
liteopt_model._save_for_lite_interpreter(lite_path) liteopt_model._save_for_lite_interpreter(lite_path)
......
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