"Fossies" - the Fresh Open Source Software Archive

Member "pytorch-1.8.2/docs/source/mobile_optimizer.rst" (23 Jul 2021, 3247 Bytes) of package /linux/misc/pytorch-1.8.2.tar.gz:


As a special service "Fossies" has tried to format the requested source page into HTML format (assuming markdown format). Alternatively you can here view or download the uninterpreted source code file. A member file download can also be achieved by clicking within a package contents listing on the according byte size field.

A hint: This file contains one or more very long lines, so maybe it is better readable using the pure text view mode that shows the contents as wrapped lines within the browser window.


torch.utils.mobile_optimizer

Warning

This API is in beta and may change in the near future.

Torch mobile supports torch.mobile_optimizer.optimize_for_mobile utility to run a list of optimization pass with modules in eval mode. The method takes the following parameters: a torch.jit.ScriptModule object, a blocklisting optimization set and a preserved method list

By default, if optimization blocklist is None or empty, optimize_for_mobile will run the following optimizations:

optimize_for_mobile will also invoke freeze_module pass which only preserves forward method. If you have other method to that needed to be preserved, add them into the preserved method list and pass into the method.

torch.utils.mobile_optimizer

optimize_for_mobile