我发现当我使用我自己的数据集(512*512)训练代码时,在第二步Fine-tune StyleGAN时会报这个错误

如果我按照网上的说法,将报错的代码后面添加strict=False

那么在下一行discriminator.load_state_dict(ckpt["d"])则会出现这个报错

我原以为可能是我数据集的原因,但是当我下载您提供的fantasy,illustration,impasto三个像素值为512*512的数据集运行时也发现了同样问题。
而我在使用您提供的cartoon数据集(1024*1024)时,它竟然能跑起来了,它们的区别仅仅是像素大小不同

这种情况可能是什么原因呢,大佬方便指导一下吗