How big is your dataset? If it one batch takes 1.5s and the whole thing takes 1h. It would mean 2400 batches. So for 10 epochs, that’s 240 batches per epoch. And so 30k samples in your dataset.
Is that correct?