ptrblck:

batch_size=128 , your DataLoader will return 10 batches à 128 samples


Thank you,
I think the confussion comes from different terminology.
I am in concrete following ResNet paper on CIFAR 10 when they have fixed the number of iterations to 64000 and batch_size to 128.
However, I think they make a distinction between iteration and batches.
Batches will be the number of batches the dataloader will throw per epoch:
Batches = Samples / Batch Size = 45000 / 128 ≈ 352
Then, the total iterations will be the batches per epoch times all the epochs:
Iterations = Epochs * Batches
Since the have fixed the number of iterations to 64000, the computation is:
Epochs = Iterations / Batches = 64000 / 352 ≈ 181
Do you agree or am I missing something?
Thanks!