TorchProfilerHparams

class composer.callbacks.callback_hparams.TorchProfilerHparams(tensorboard_trace_handler_dir: str = 'torch_profiler', tensorboard_use_gzip: bool = False, record_shapes: bool = True, profile_memory: bool = False, with_stack: bool = True, with_flops: bool = True, skip: int = 0, warmup: int = 1, active: int = 5, wait: int = 0, repeat: int = 0)

Bases: composer.callbacks.callback_hparams.CallbackHparams

TorchProfiler hyperparameters.

See TorchProfiler for documentation.

initialize_object() → TorchProfiler

Initialize the callback.

Returns

Callback – An instance of the callback.