1. Some images are not 224 * 224 large, directly transforms.RandomCrop (224) will error, RandomRange error
raise ValueError("empty range for randrange() (%d,%d, %d)" % (istart, istop, width)) ValueError
You need to add transform.Resize ((256,256))
transforms.Resize((256, 256)), transforms.RandomCrop(224),