composer.profiler.dataloader_profiler#

Profiler to measure the time it takes the data loader to return a batch.

Classes

DataLoaderProfiler

Profile a DataLoader.
class composer.profiler.dataloader_profiler.DataLoaderProfiler[source]#

Bases: composer.core.callback.Callback

Profile a DataLoader.

This callback measures the latency it takes for the DataLoader to yield a batch.

Note

The Composer Trainer automatically creates an instance of this DataLoaderProfiler callback whenever the profiler is enabled.

When using the Composer Trainer, one does not need to directly create an instance of this DataLoaderProfiler callback.