hkchengrex/Cutie

Tips on training on single GPU

huyhoan1109 opened this issue · 1 comments

I get crashed every times when I trained your model in single GPU (NVIDIA GeForce RTX 3090 24GB) due to insufficient memory. Can you give me some tips to fix that.

You would need either a GPU with more memory or more GPUs.