Run python trainer.py fit --trainer.max_epochs=30 --data.batch_size=6 locally.
Related Issues
Note that the shuffling operation is slower than in-batch shuffling. There is a longer delay at the start as the image chips are added to the shuffle buffer, and each mini-batch is now processing about 2x slower (one iteration used to take ~1s, now it takes ~2s).
What I am changing
How I did it
.shuffle
instead of.in_batch_shuffle
How you can test it
python trainer.py fit --trainer.max_epochs=30 --data.batch_size=6
locally.Related Issues
Note that the shuffling operation is slower than in-batch shuffling. There is a longer delay at the start as the image chips are added to the shuffle buffer, and each mini-batch is now processing about 2x slower (one iteration used to take ~1s, now it takes ~2s).