实验环境中pytorch版本是0.4.1,torchvision版本是0.2.1。(这版本看着就眼熟是不是)
报了错误:
TypeError:an integer is required (got type tuple)
因为错误上下有经过 torch.transform
处理,所以想的是这里有错误。果不其然,搜到了大兄弟下面的博客。原来是 RandomResizedCrop
这里出了问题,把
transforms.RandomResizedCrop((222, 222), (0.8, 1))
改为
transforms.RandomResizedCrop(222, (0.8, 1))
就好了~