Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
lixiaoyouran opened this issue Apr 15, 2024 · 10 comments
Open

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

lixiaoyouran opened this issue Apr 15, 2024 · 10 comments

Comments

@lixiaoyouran
Copy link

我发现当我使用我自己的数据集(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
这种情况可能是什么原因呢,大佬方便指导一下吗

@williamyang1991
Copy link
Owner

StyleGAN每种分辨率是独立的模型,你需要专门去下载512参数的预训练StyleGAN,比如
https://github.com/justinpinkney/awesome-pretrained-stylegan2?tab=readme-ov-file#faces-FFHQ-config-f-512x512

@lixiaoyouran
Copy link
Author

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.*版本

image

下载tensorflow2.0以上的版本后就有这个报错,网上说要降低版本

image

@williamyang1991
Copy link
Owner

我的模型不是tensorflow的版本,是pytorch的版本。

@williamyang1991
Copy link
Owner

https://github.com/rosinality/stylegan2-pytorch?tab=readme-ov-file#convert-weight-from-official-checkpoints
这个。他提供了把tensorflow的版本转成pytorch的代码

@lixiaoyouran
Copy link
Author

https://github.com/rosinality/stylegan2-pytorch?tab=readme-ov-file#convert-weight-from-official-checkpoints 这个。他提供了把tensorflow的版本转成pytorch的代码

好的,我试试,谢谢大佬

@lixiaoyouran
Copy link
Author

https://github.com/rosinality/stylegan2-pytorch?tab=readme-ov-file#convert-weight-from-official-checkpoints 这个。他提供了把tensorflow的版本转成pytorch的代码

大佬,我想问一下,您在论文里写的
image

这几句话是什么意思,方便回答且举一下例子吗,还是小白,有点困惑,特别是第一句,Toonify, FS-Ada and U-GAT-IT learn domain-level rather than image-level styles.是什么意思呢,domain-level和image-level 的不同体现在什么方面

@williamyang1991
Copy link
Owner

他们学的是一个数据集的整体风格,我们的方法学的是一个数据集里每张图片的风格。

@lixiaoyouran
Copy link
Author

他们学的是一个数据集的整体风格,我们的方法学的是一个数据集里每张图片的风格。

奥奥, 大佬还可以举例一下这两类的几个算法吗,实际人脸卡通化中,好像domain-level使用得比较多,各类视频软件软件简单的某种风格迁移应该都属于这类😭,dualStyleGan这种可以设置id的有对应风格映射来生成的还有类似的算法吗,谢谢大佬了😚

@williamyang1991
Copy link
Owner

我论文里写了。StarGAN2和GNR

@lixiaoyouran
Copy link
Author

我论文里写了。StarGAN2和GNR

啊,不好意思,还没细看到那里,打扰了,谢谢啦

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants