调试StarGAN v2 代码遇到的一个大坑

这两天想用一下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