search

williamyang1991 / DualStyleGAN

[CVPR 2022] Pastiche Master: Exemplar-Based High-Resolution Portrait Style Transfer
Other
1.61k stars 249 forks source link

加载预训练模型出错 #104

Closed Xia-zx closed 3 months ago

Xia-zx commented 3 months ago
微信图片_20240529185822 1 2

什么原因，求助

williamyang1991 commented 3 months ago

你提供的信息过少，我没法帮你

Xia-zx commented 3 months ago

你提供的信息过少，我没法帮你

我按照readme中说的，下载了cartoon的generator.pt按照要求目录存放，运行python style_transfer.py ，在76行generator.load_state_dict(ckpt["g_ema"])，出现报错

williamyang1991 commented 3 months ago

至少提供你输入的命令行吧

Xia-zx commented 3 months ago

python style_transfer.py

完全按照readme中说的，命令行就是python style_transfer.py ，没提供参数

williamyang1991 commented 3 months ago

那我只能猜测你模型下载错了 你可以试着下载一下其他风格的，比如caricature，然后运行一下

python style_transfer.py --style caricature --name caricature_transfer --style_id 0 --weight 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1

看是不是报一样的错。 如果没报错就是你下载的有问题 如果也报错，那可能是代码哪里有问题（但是那么多人都试过了一直都没啥问题），你可以试一下添加

generator.load_state_dict(ckpt["g_ema"], strict=False)

去忽略那些unexpected key

Xia-zx commented 3 months ago

那我只能猜测你模型下载错了 你可以试着下载一下其他风格的，比如caricature，然后运行一下

python style_transfer.py --style caricature --name caricature_transfer --style_id 0 --weight 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1

看是不是报一样的错。 如果没报错就是你下载的有问题 如果也报错，那可能是代码哪里有问题（但是那么多人都试过了一直都没啥问题），你可以试一下添加

generator.load_state_dict(ckpt["g_ema"], strict=False)

去忽略那些unexpected key

添加

generator.load_state_dict(ckpt["g_ema"], strict=False)

后就好了，感谢帮助