About GPU memory - Githubissues

exitudio / MMM

Official repository for "MMM: Generative Masked Motion Model"

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

79 stars 5 forks source link

Open xlbjy opened 5 months ago

xlbjy commented 5 months ago

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

exitudio commented 5 months 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.