Back to Pytorch

Samplers

docs/cpp/source/api/data/samplers.md

2.12.01.2 KB
Original Source

Samplers

Samplers control the order in which samples are accessed from a dataset. They determine the indices that the DataLoader uses to fetch data.

Sampler Base Class

{doxygenclass}
:members:
:undoc-members:

Sequential Sampler

Accesses samples in order from 0 to N-1. Use this for evaluation or when order matters.

{doxygenclass}
:members:
:undoc-members:

Random Sampler

Accesses samples in random order. Use this for training to ensure the model sees samples in different orders each epoch.

{doxygenclass}
:members:
:undoc-members:

Distributed Random Sampler

For distributed training, ensures each process gets a different subset of the data without overlap.

{doxygenclass}
:members:
:undoc-members:

Distributed Sampler (Base)

{doxygenclass}
:members:
:undoc-members:

Distributed Sequential Sampler

{doxygenclass}
:members:
:undoc-members:

Stream Sampler

{doxygenclass}
:members:
:undoc-members: