accelerate
3d1a0f7e - fix attribute error in DataloaderShared (#1278)

Commit
2 years ago
fix attribute error in DataloaderShared (#1278) When running in single GPU, the `batch_sampler` of `DataLoaderShared` is a `torch.utils.data.sampler.BatchSampler` object instead of `DataSamplerShared ` object, which does not contain necessary attributes to calculate `total_batch_size`.
Author
Parents
Loading