Open prashantg445 opened 4 years ago
Sorry for not replying earlier. @prashantg445 might be difficult to tell from nvidia-smi whether the GPU is fully utilized or not. Try to scale down the images to 128x128 and see if it is gonna work or not. What is your image resolution by the way? More, torch==1.4.0 should work immediately.
I had 7 GB GPU VRAM available and around 13 GB RAM available with training configurations:
I have placed my training data outside the repo with relative path (as per cyclegan_text.py):
../data//train/A/
../data//train/B/
Similarly test files. Also, commented line 56 in cyclegan_text.py:
#############
But when I run python3 cyclegan_text.py, I get following error trace:
I am watching nvidia-smi but it seems that GPU is not being fully utilized, neither by this process nor by some other processes.
Tried, but didn't worked: