Closed Xia-zx closed 3 months ago
你提供的信息过少,我没法帮你
你提供的信息过少,我没法帮你
我按照readme中说的,下载了cartoon的generator.pt按照要求目录存放,运行python style_transfer.py ,在76行generator.load_state_dict(ckpt["g_ema"]),出现报错
至少提供你输入的命令行吧
python style_transfer.py
完全按照readme中说的,命令行就是python style_transfer.py ,没提供参数
那我只能猜测你模型下载错了 你可以试着下载一下其他风格的,比如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
那我只能猜测你模型下载错了 你可以试着下载一下其他风格的,比如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)
后就好了,感谢帮助
什么原因,求助