API ReferenceOther APIsAituneTorchInspecting

aitune.torch.inspecting.wrapping

View as Markdown

Inspecting and patching modules.

Module Contents

Functions

NameDescription
wrapWrap provided modules with inspection logic.

Data

LOG_FORMAT

logger

API

aitune.torch.inspecting.wrapping.wrap(
    obj: object,
    modules: list[aitune.torch.inspecting.module_info.ModuleInfo],
    strategy: aitune.torch.tune_strategy.tune_strategy.TuneStrategy | None = None,
    strategies: aitune.torch.module.wrapper_module.StrategyList | aitune.torch.module.wrapper_module.StrategyMap | None = None
) -> object

Wrap provided modules with inspection logic.

Parameters:

obj
object

Callable object to wrap.

modules
list[ModuleInfo]

Dictionary of module names and their corresponding ModuleInfo objects.

strategy
TuneStrategy | NoneDefaults to None

Strategy to use for patching.

strategies
StrategyList | StrategyMap | NoneDefaults to None

Strategies to use for patching.

Returns: object

Wrapped callable object.

aitune.torch.inspecting.wrapping.LOG_FORMAT = '%(asctime)s - %(levelname)s - %(message)s'
 
aitune.torch.inspecting.wrapping.logger = getLogger(__name__)