RuntimeError: Error(s) in loading state_dict for Discriminator:
lixiaoyouran opened this issue · comments
我发现当我使用我自己的数据集（512*512）训练代码时，在第二步Fine-tune StyleGAN时会报这个错误
如果我按照网上的说法，将报错的代码后面添加strict=False
那么在下一行discriminator.load_state_dict(ckpt["d"])则会出现这个报错
我原以为可能是我数据集的原因，但是当我下载您提供的fantasy,illustration,impasto三个像素值为512*512的数据集运行时也发现了同样问题。
而我在使用您提供的cartoon数据集(1024*1024)时，它竟然能跑起来了，它们的区别仅仅是像素大小不同
这种情况可能是什么原因呢，大佬方便指导一下吗
StyleGAN每种分辨率是独立的模型，你需要专门去下载512参数的预训练StyleGAN，比如
https://github.com/justinpinkney/awesome-pretrained-stylegan2?tab=readme-ov-file#faces-FFHQ-config-f-512x512
StyleGAN每种分辨率是独立的模型，你需要专门去下载512参数的预训练StyleGAN，比如 https://github.com/justinpinkney/awesome-pretrained-stylegan2?tab=readme-ov-file#faces-FFHQ-config-f-512x512
大佬，就是您发的这个StyleGAN这个分辨率的模型，我下载后运行代码，发现tensorflow报错，您的代码是Python3.8版本的，只支持tensorflow2.0以上，不能下载tensorflow1.几的版本，但是这个StyleGan2的模型tensorflow1.几的版本，我还是运行不了。我目前也不知道怎么实现。
而且我找了一些stylegan2的人脸预训练模型，都是好几年前的，都是旧的版本。这个怎么解决呢?
python3.8下载不了tensorflow1.*版本
下载tensorflow2.0以上的版本后就有这个报错，网上说要降低版本
我的模型不是tensorflow的版本，是pytorch的版本。
https://github.com/rosinality/stylegan2-pytorch?tab=readme-ov-file#convert-weight-from-official-checkpoints
这个。他提供了把tensorflow的版本转成pytorch的代码
https://github.com/rosinality/stylegan2-pytorch?tab=readme-ov-file#convert-weight-from-official-checkpoints 这个。他提供了把tensorflow的版本转成pytorch的代码
好的，我试试，谢谢大佬
https://github.com/rosinality/stylegan2-pytorch?tab=readme-ov-file#convert-weight-from-official-checkpoints 这个。他提供了把tensorflow的版本转成pytorch的代码
大佬，我想问一下，您在论文里写的
这几句话是什么意思，方便回答且举一下例子吗，还是小白，有点困惑，特别是第一句，Toonify, FS-Ada and U-GAT-IT learn domain-level rather than image-level styles.是什么意思呢，domain-level和image-level 的不同体现在什么方面
他们学的是一个数据集的整体风格，我们的方法学的是一个数据集里每张图片的风格。
他们学的是一个数据集的整体风格，我们的方法学的是一个数据集里每张图片的风格。
奥奥， 大佬还可以举例一下这两类的几个算法吗，实际人脸卡通化中，好像domain-level使用得比较多，各类视频软件软件简单的某种风格迁移应该都属于这类😭，dualStyleGan这种可以设置id的有对应风格映射来生成的还有类似的算法吗，谢谢大佬了😚
我论文里写了。StarGAN2和GNR
我论文里写了。StarGAN2和GNR
啊，不好意思，还没细看到那里，打扰了，谢谢啦