search

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.68k stars 306 forks source link

Batch size #220

Closed HuygheB closed 2 years ago

HuygheB commented 2 years ago

Hi guys, how can I change the batch size so I can accelerate training times?

songyangsun commented 2 years ago

batchsize = gpu nums * imgs_per_gpu , which in solov2_r50_fpn_8gpu_3x.py