If the number of images is smaller than the batch size, you’ll only get these images (in your case two images).
yep now getting it perfect…
torch.Size([32, 3, 32, 32])
torch.Size([32, 3, 32, 32])
torch.Size([32, 3, 32, 32])
torch.Size([32, 3, 32, 32])
torch.Size([32, 3, 32, 32])
but in between I am getting below warning as well…
/anaconda3/lib/python3.6/site-packages/PIL/TiffImagePlugin.py:763: UserWarning: Possibly corrupt EXIF data. Expecting to read 206 bytes but only got 0. Skipping tag 0
" Skipping tag %s" % (size, len(data), tag))
/anaconda3/lib/python3.6/site-packages/PIL/TiffImagePlugin.py:763: UserWarning: Possibly corrupt EXIF data. Expecting to read 65536 bytes but only got 0. Skipping tag 0
" Skipping tag %s" % (size, len(data), tag))
/anaconda3/lib/python3.6/site-packages/PIL/TiffImagePlugin.py:763: UserWarning: Possibly corrupt EXIF data. Expecting to read 170 bytes but only got 0. Skipping tag 0
" Skipping tag %s" % (size, len(data), tag))
/anaconda3/lib/python3.6/site-packages/PIL/TiffImagePlugin.py:763: UserWarning: Possibly corrupt EXIF data. Expecting to read 178 bytes but only got 0. Skipping tag 0
" Skipping tag %s" % (size, len(data), tag))
/anaconda3/lib/python3.6/site-packages/PIL/TiffImagePlugin.py:763: UserWarning: Possibly corrupt EXIF data. Expecting to read 278 bytes but only got 0. Skipping tag 0
" Skipping tag %s" % (size, len(data), tag))
/anaconda3/lib/python3.6/site-packages/PIL/TiffImagePlugin.py:780: UserWarning: Corrupt EXIF data. Expecting to read 12 bytes but only got 11.
warnings.warn(str(msg))