torchsig.utils.data_loadingΒΆ

Collate function and DataLoader with worker seeding for TorchSig. Provides:

  • metadata_padding_collate_fn: pads variable-length metadata in each batch.

  • WorkerSeedingDataLoader: seeds each worker process differently for reproducibility.

Functions

metadata_padding_collate_fn

Collate a batch of (data, metadata_list) pairs, padding metadata to equal lengths.

Classes

WorkerSeedingDataLoader

DataLoader that seeds each worker process differently using a shared seed.