GPU Memory needed for training

Question

GPU Memory needed for training

Closed this issue a year ago · 2 comments

Thank you for your interesting work.
I have a question regarding the GPU memory size you used for training.
I tried to run disentangle.py with an NVIDIA RTX-3090 GPU with 24GB of memory and got an out-of-memory error.

Thank you in advance,
Thodoris

Answer 1 · 2023-10-23T18:39:43.000Z

Hi Thodoris,

Thank you very much for your interest to our work! Unfortunately, as described here, this project requires a GPU memory of 48GB. You might want to try with a larger GPU. Sorry for the inconvenience!

Sincerely,
Qiucheng

Answer 2 · 2023-10-25T12:18:04.000Z

Hello Qiucheng,
I was looking for that info in the paper and didn't look at your README.
Thank you very much!
Thodoris