search

mingyuliutw / UNIT

Unsupervised Image-to-Image Translation
Other
1.98k stars 360 forks source link

Testing - Multi GPU #91

Closed mhusseinsh closed 5 years ago

mhusseinsh commented 5 years ago

I am trying to test my trained model on an image with [720, 1280] and I run out of memory. My GPU is P100 with 16GB, but still runs out of memory after around 6 images

How can I used multiple GPUs in the test phase ?

mingyuliutw commented 5 years ago

Please check #93