cascade-rcnn icon indicating copy to clipboard operation
cascade-rcnn copied to clipboard

Questions about multi-gpu and multiple batch size

Open yaoqi-zd opened this issue 6 years ago • 2 comments

I haven't run caffe with multiple gpus before, so I'm a little confused that why the batch size can set larger than 1 in Cascade R-CNN given that the input images are not resized to exactly the same size? As far as I know, the image size should equal in a single batch.

yaoqi-zd avatar Jul 25 '18 07:07 yaoqi-zd

Each gpu is running independently during forward and backward, but the gradients for weights will be aggregated before updating. So the image sizes could be different across GPUs.

zhaoweicai avatar Aug 13 '18 04:08 zhaoweicai

Does this setup support batch training (batch > 1) for a single GPU?

ramanathan831 avatar Feb 24 '21 23:02 ramanathan831