spdl.source

Overview

Iterables for traversing datasets and utilities for transforming them.

API Reference

Classes

IterableWithShuffle

IterableWithShuffle

A protocol that is often used to represent data source.

SizedIterable

SizedIterable

A protocol that is often used to represent data source.

SizedIterableWithShuffle

SizedIterableWithShuffle

A protocol that is often used to represent data source.

DistributedRandomSampler

DistributedRandomSampler

Sample dataset indices for the given distributed node while applying randomness.

DistributedDeterministicSampler

DistributedDeterministicSampler

Sampler for distributed training that splits indices across multiple ranks.

Modules

imagenet

imagenet

Iterable for ImageNet dataset.

local_directory

local_directory

Iterable for traversing local file directory.

utils

utils

Implements meta-transformations on iterables/iterators.