search

chenhaoxing / DiffusionInst

This repo is the code of paper "DiffusionInst: Diffusion Model for Instance Segmentation" (ICASSP'24).
Apache License 2.0
221 stars 12 forks source link

Lower GPU memory #22

Closed VLadImirluren closed 5 months ago

VLadImirluren commented 10 months ago

Thanks to your great work!

I have only 2 A100 GPUs.

Could you please give me some advice about reproducing this wonderful work?

zhangxgu commented 5 months ago

Hi, thanks for your attention. If you only have 2 A100, we suggest you to use bs/4 and lr/2 for training since our experiments are done on 8 a100s