update dataloader wrappers to have `total_batch_size` attribute (#493)
* update dataloader wrappers to have `total_batch_size` attribute
* fix
* resolving comments
* Update src/accelerate/data_loader.py
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
* quality
* add docstrings
* Update src/accelerate/data_loader.py
Co-authored-by: Zachary Mueller <muellerzr@gmail.com>
* docstrings iter 2 + quality + minor change in doc
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Zachary Mueller <muellerzr@gmail.com>