加载预训练模型出错 about dualstylegan HOT 6 CLOSED

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

from dualstylegan.

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

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

from dualstylegan.

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

from dualstylegan.

python style_transfer.py

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

from dualstylegan.

那我只能猜测你模型下载错了
你可以试着下载一下其他风格的，比如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

from dualstylegan.

那我只能猜测你模型下载错了 你可以试着下载一下其他风格的，比如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)

后就好了，感谢帮助

from dualstylegan.