search

chavinlo / musicgen_trainer

simple trainer for musicgen/audiocraft
GNU Affero General Public License v3.0
13 stars 1 forks source link

Any idea on vram needed for medium, large training? #11

Open DavidNTompkins opened 1 year ago

DavidNTompkins commented 1 year ago

I've gotten small to train but can't get medium+ to train even at batch=1 on an A100. I can't tell if I'm just being dumb or if this needs huge vram