Giters

AlamiMejjati / Unsupervised-Attention-guided-Image-to-Image-Translation

Unsupervised Attention-Guided Image to Image Translation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training on GPU is taking a lot of RAM

viniciusarruda opened this issue · comments

commented

I'm training on a GPU but it's taking a lot of RAM (I have 24 GB and it's consuming ~22GB). Is this expected ? What can I do to reduce this amount of RAM consumption ?

commented

Hello vinicius, I have the same memory consumption as you (around 22GB). I am not sure yet why the amount of RAM required is that high but I will get back to you when I figure it out. Meanwhile maybe you can try reducing the number of residual blocks from the generators.