For IterableDataset, return DataLoader using self._train_batch_size. …
…This is consistent with how we generate a regular DataLoader, and leads to the correct args.per_device_train_batch_size eventually ending up on each GPU.

Fixes # 21444#issuecomment-1416252207

@sgugger 
_The documentation is not available anymore as the PR was closed or merged._