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

最后一个test_painterly_image...这个程序runtimeError,在哪里更改部分参数呢 #5

Closed
luojunqi011027 opened this issue Nov 20, 2023 · 1 comment

Comments

@luojunqi011027
Copy link

IMG_3848

@bo-zhang-cs
Copy link
Collaborator

这是显存不足导致的错误。如果有多张显卡,可以将test_painterly_image_harmonization.pyphd_net, phdiffphdiffres这三个模型的device设成不同的值,将不同模型放在不同显卡上;如果只有一张卡,可以注释其他模型的代码,每次只测试一个模型,以此减小显卡占用。

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