# 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(self, steps_to_profile=5)

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