About GPU memory - Githubissues

exitudio / MMM

Official repository for "MMM: Generative Masked Motion Model"

https://exitudio.github.io/MMM-page/

46 stars 1 forks source link

Open xlbjy opened 3 weeks ago

xlbjy commented 3 weeks ago

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

exitudio commented 3 weeks ago

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