I was trying to train on all data so I used split_none(). The train dataset in my databunch has 3000 items, which is as expected. I then created and trained a model, which seemed to be working as expected. But when I try to get the predictions…
I know this is old, but for the sake of others like me…
It happened to me as well. As @kushaj indicated the last batch is being ignored.
My dataset size was 15620 rows and
15620 % 64 (the default batch size) = 4 --> Hence I was not able to predict for the last batch of size 4
To get around it, I found the prime factors and took the biggest product of these factors that I could afford to compute quickly
Positive Integer factors of 15620 = 2, 4, 5, 20, 11, 220, 71