How much memory do I need for trainning?

Open edith-wq opened this issue 1 year 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.

Mar 25 '24 00:03 edith-wq

Hi, Can you share the training code and define the methodologies used?

Mar 25 '24 04:03 pushkarraj

Hi, Can you share the training code and define the methodologies used?

I try to write a trainning scrpt ,u can refer this code https://github.com/lyc0929/OOTDiffusion-train.

Mar 29 '24 12:03 lyc0929