search

bcmi / libcom

Image composition toolbox: everything you want to know about image composition or object insertion
Apache License 2.0
276 stars 18 forks source link

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

Closed luojunqi011027 closed 10 months ago

luojunqi011027 commented 10 months ago

IMG_3848

bo-zhang-cs commented 10 months ago

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