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.