For example, If I have a dataset with 10 rows. I want to train an MLP/RNN/CNN on this using mini batches.
So, let’s say, I take 2 rows at a time to train. 2 x 5 = 10. So, I train my model with batches where each batch contains 2 rows. So, number of batches = 5 and number of rows per batch is 2.
Is my batch_size 2? or is it 5?
In the dataloader, should my batch_size = 2 or batch_size = 5?