Video memory requirement

Open WusterHappy opened this issue 2 years ago • 3 comments

May I ask how much video memory your project needs at least to run? Is 24g 3090 enough

May 09 '23 07:05 WusterHappy

Good question, 24 GB memory is enough for all of the models. If you use the vit-b, 12GB memory can run the model.

May 11 '23 00:05 Qsingle

Is it enough? I am trying to train the vit-b in 24gb GPU model and it only train in bs 1

Jun 23 '23 20:06 krishnaadithya

Yeah, it is enough. For the ViT-H, bs 1 is the largest batch size we can run on one GPU. You can try other backbone like ViT-B or ViT-L

Jul 02 '23 02:07 Qsingle