\'\'\'
imsize = 512 if torch.cuda.is_available() else 128 loader = transforms.Compose([transforms.Resize(imsize),transforms.ToTensor()]) def image_loader(i