mode — Model Optimizer 0.27.1 (original) (raw)
Module implementing and describing modes that can be used during the NAS convert process.
Check out mtn.convert to learn more about modes.
Classes
| AutoNASModeDescriptor | Class to describe the "autonas" mode. |
|---|---|
| ExportModeDescriptor | Class to describe the "export" mode. |
class AutoNASModeDescriptor
Bases: _ModeDescriptor
Class to describe the "autonas" mode.
The properties of this mode can be inspected via the source code.
property config_class_: type[ModeloptBaseConfig]_
Specifies the config class for the mode.
property convert_: Callable[[Module, ModeloptBaseConfig], tuple[Module, dict[str, Any]]]_
The mode’s entrypoint for converting a model.
property export_mode_: str | None_
The mode that corresponds to the export mode of this mode.
property name_: str_
Returns the value (str representation) of the mode.
property next_modes_: set[str] | None_
Modes that must immediately follow this mode.
property restore_: Callable[[Module, ModeloptBaseConfig, dict[str, Any]], Module]_
The mode’s entrypoint for restoring a model.
property search_algorithm_: type[BaseSearcher]_
Specifies the search algorithm to use for this mode (if any).
property update_for_new_mode_: Callable[[Module, ModeloptBaseConfig, dict[str, Any]], None]_
The mode’s entrypoint for updating the models state before new mode.
property update_for_save_: Callable[[Module, ModeloptBaseConfig, dict[str, Any]], None]_
The mode’s entrypoint for updating the models state before saving.
class ExportModeDescriptor
Bases: _ModeDescriptor
Class to describe the "export" mode.
The properties of this mode can be inspected via the source code.
property config_class_: type[ModeloptBaseConfig]_
Specifies the config class for the mode.
property convert_: Callable[[Module, ModeloptBaseConfig], tuple[Module, dict[str, Any]]]_
The mode’s entrypoint for converting a model.
property is_export_mode_: bool_
Whether the mode is an export mode.
Returns:
True if the mode is an export mode, False otherwise. Defaults to False.
property name_: str_
Returns the value (str representation) of the mode.
property restore_: Callable[[Module, ModeloptBaseConfig, dict[str, Any]], Module]_
The mode’s entrypoint for restoring a model.