OOTDiffusion icon indicating copy to clipboard operation
OOTDiffusion copied to clipboard

How much memory do I need for trainning?

Open edith-wq opened this issue 11 months ago • 2 comments

How much GPU memory do I need for trainning? I have A100 x 8 (40G). I got CUDA oom error when i try trainning. ( i wrote own trainning code ) I use huggingface acclerate.

edith-wq avatar Mar 25 '24 00:03 edith-wq