batch size details

[hdjang]

Hi

I'd like to ask a few details on batch size in two-step training.

stage1:
If I set BATCH_SIZE_IMG=128 in config file, then using 4 gpu would match 512 batch size of stage1 as noted in the paper. Is this right?

stage2:
This stage is hard for me to set batch size. How should I set BATCH_SIZE_3D: ?, BATCH_SIZE_2D: ?, BATCH_SIZE_IMG: ? to match 32 batch size of stage2 as noted in the paper?

Thanks