DDP icon indicating copy to clipboard operation
DDP copied to clipboard

about gpu memory and batch size

Open filbertsun opened this issue 1 year ago • 1 comments

I encountered a confusing problem when I tried to train the depth model. When I executed bash tools/dist_train.sh configs/ddp_kitti/ddp_swinb_22k_w7_kitti_bs2x8_scale01.py 1 with samples_per_gpu=2, the model took the following GPU memory. image However, after I set samples_per_gpu=4, the occupied memory was reduced. image Why would that happen?

filbertsun avatar Oct 30 '23 09:10 filbertsun