search

ericsujw / InstColorization

MIT License
712 stars 152 forks source link

Running on CPU #13

Closed boblef closed 4 years ago

boblef commented 4 years ago

Is there any way to run the script on a machine that doesn't have GPUs?

jerrymatjila commented 4 years ago

Have a look at pull request https://github.com/ericsujw/InstColorization/pull/12 I have tested it by running the example with option --gpu_ids -1 Here is the full command

python test_fusion.py --name test_fusion --sample_p 1.0 --model fusion --test_img_dir example --results_img_dir results --gpu_ids -1
boblef commented 4 years ago

Thank you!