# utils.callbacks.profiler – Axolotl

# utils.callbacks.profiler

`utils.callbacks.profiler`

HF Trainer callback for creating pytorch profiling snapshots

## Classes

Name | Description
--- | ---
PytorchProfilerCallback | PyTorch Profiler callback to create snapshots of GPU memory usage at specified steps.

### PytorchProfilerCallback

    utils.callbacks.profiler.PytorchProfilerCallback(
        steps_to_profile=5,
        profiler_steps_start=0,
    )

PyTorch Profiler callback to create snapshots of GPU memory usage at specified steps.