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.