Data loader. Combines a dataset and a sampler, and provides an iterable over
train_loader()loader
the given dataset. The :class:`~torch.utils.data.DataLoader` supports both map-style and iterable-style datasets with single- or multi-process loading, customizing loading order and optional automatic batching (collation) and memory pinning.