OpenGVLab/SAM-Med2D

batchsize的大小

21-10-4 opened this issue · 2 comments

21-10-4 commented

SAM-Med2D的论文提到：“8 A100(80G)，仅仅微调了the base model of SAM”
如果采用256*256的分辨率，我感觉batchsize可以设置的非常大；我在自己的数据集上微调效果很差，所以想知道作者使用的batchsize的大小

Grace-yafei commented

我也遇到了相同的情况，3090（24G）上可以设置batch_size为32甚至更大，微调效果同样很差，请问这种情况是正常的吗

Cjl-MedSeg commented

In our paper, we mentioned that each GPU handles 50 images and their corresponding 250 masks. Therefore, we set the batch size to 50 during training.