May I ask how much VRAM is required to train the segmentation model?

fudan-zvg / meta-prompts

MIT License

69 stars 2 forks source link

May I ask how much VRAM is required to train the segmentation model? #12

Open JKL-uxf opened 6 months ago

JKL-uxf commented 6 months ago

I tried this model with 3080TI, but I've faced OOM problem. Even with 3080TI with 3060TI with Deepspeed + zero 3 approach, I've faced OOM issue again.

So how much Vram memory does this segmentation model need?

wwqq commented 6 months ago

Hi @JKL-uxf 40G needed

JKL-uxf commented 6 months ago

Thank you for the answer.

JKL-uxf commented 6 months ago

@wwqq I am sorry to ask you again, May I ask you how much time to train cityscapes dataset? I am planning to subscribe cloud computing service.

water221 commented 1 month ago

I only have a 3090,24G video memory card now,Can you adjust some parameters to run the training task of depth estimation?,If you can,Please reply.,In addition, how long did it take to train on your V100 card?