这两天想用一下StarGAN v2来完成自己的一个任务,由于本人的Windows配置比实验室服务器好用,所以本人习惯于用Windows来跑程序。对于pytorch来说,windows和服务器还是有一丢丢的差别的。
调试的时候,总是出现 AttributeError: Can’t pickle local object “get_train_loader< locals><\lambda>"。在网上搜了很多资料,都没有搜到,最后在https://zhiqianghe.blog.csdn.net/article/details/110549270这里给出的那个连接李找到了思路。因为本人是半路出家,所以对lambda函数不熟悉。但好在最终还是找到了问题所在。问题在于get_train_loader函数里面有下面语句”
rand_crop = transforms.Lambda(
lambda x: crop(x) if random.random() < prob else x)
其中:
crop = transforms.RandomResizedCrop(
img_size, scale=[0.8, 1.0], ratio=[0.9, 1.1])
我解决的方法是把rand_crop去掉lambda模块,将他修改成函数。也就是将 transforms中的RandomResizedCrop函数复制出来,修改一下即可。即将
transform = transforms.Compose([
rand_crop,
transforms.Resize([img_size, img_size]),
transforms.RandomHorizontalFlip(),
transforms.ToTenso