Yes, more or less. It will actually give you 33 batches. 32 batches of size 15 and 1 batch of size 1, as the number of samples is not dividable by 15 without a remainder and you didn’t specify drop_last=True in your DataLoader.