MMM icon indicating copy to clipboard operation
MMM copied to clipboard
verified publisher icon exitudio

About GPU memory

Open xlbjy opened this issue 1 year ago • 1 comments

Hi, nice work. Is one Nvidia RTX 3090 GPU (24GB) enough for training?

xlbjy avatar Jun 12 '24 09:06 xlbjy

Hi, thank you for your interest. It should be fine. It uses around ~17 GB, 1GPU, with batchsize 128 (in 2nd stage) for training.

exitudio avatar Jun 12 '24 16:06 exitudio