@Subhobrata_Mukharjee I ran your note book after some debugging seems like there is issue with some of your image files encodings. I didn’t debug enough to figure out which ones but that should be a clue. PIL seems to throw that error in between training so, that’s another point towards image encoding errors.