A detailed example of data loaders with PyTorch
stanford.edu › ~shervine › blogPyTorch script Now, we have to modify our PyTorch script accordingly so that it accepts the generator that we just created. In order to do so, we use PyTorch's DataLoader class, which in addition to our Dataset class, also takes in the following important arguments: batch_size, which denotes the number of samples contained in each generated batch.