Tools

FrozenClass

class catalyst.tools.frozen_class.FrozenClass

Bases: object

Class which prohibit __setattr__ on existing attributes.

Examples

>>> class IRunner(FrozenClass):

MetricHandler

class catalyst.tools.metric_handler.MetricHandler(minimize: bool = True, min_delta: float = 1e-06)

Bases: object

@TODO: docs.

Parameters

• minimize – @TODO: docs

• min_delta – @TODO: docs

ModelForwardWrapper

class catalyst.tools.forward_wrapper.ModelForwardWrapper(model, method_name)

Bases: torch.nn.modules.module.Module

Model that calls specified method instead of forward.

Parameters

• model – @TODO: docs

• method_name – @TODO: docs

(Workaround, single method tracing is not supported)

forward(*args, **kwargs)

Forward pass.

Parameters

• *args – some args

• **kwargs – some kwargs

Returns

specified method output

Return type

output

TimeManager

class catalyst.tools.time_manager.TimeManager

Bases: object

@TODO: Docs. Contribution is welcome.

reset() → None

Reset all previous timers.

start(name: str) → None

Starts timer name.

Parameters

name – name of a timer

stop(name: str) → None

Stops timer name.

Parameters

name – name of a timer