Skip to content

RuntimeError: Error(s) in loading state_dict for Discriminator: #100

@lixiaoyouran

Description

@lixiaoyouran

我发现当我使用我自己的数据集(512*512)训练代码时,在第二步Fine-tune StyleGAN时会报这个错误
image
如果我按照网上的说法,将报错的代码后面添加strict=False
image

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

我原以为可能是我数据集的原因,但是当我下载您提供的fantasy,illustration,impasto三个像素值为512*512的数据集运行时也发现了同样问题。

而我在使用您提供的cartoon数据集(1024*1024)时,它竟然能跑起来了,它们的区别仅仅是像素大小不同
image
这种情况可能是什么原因呢,大佬方便指导一下吗

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions