Skip to content

DataLoaders、Optimizers 和 Schedulers

Accelerate 在调用 [~Accelerator.prepare] 时用于准备对象以进行分布式训练的内部类。

DataLoader 工具

[[autodoc]] data_loader.prepare_data_loader [[autodoc]] data_loader.skip_first_batches

BatchSamplerShard

[[autodoc]] data_loader.BatchSamplerShard

IterableDatasetShard

[[autodoc]] data_loader.IterableDatasetShard

DataLoaderShard

[[autodoc]] data_loader.DataLoaderShard

DataLoaderDispatcher

[[autodoc]] data_loader.DataLoaderDispatcher

AcceleratedOptimizer

[[autodoc]] optimizer.AcceleratedOptimizer

AcceleratedScheduler

[[autodoc]] scheduler.AcceleratedScheduler