GoogleConceptualCaptioning icon indicating copy to clipboard operation
GoogleConceptualCaptioning copied to clipboard

Multi-GPU instructions

Open sgondala opened this issue 4 years ago • 3 comments

How do I train the code over multi-GPU setup?

sgondala avatar May 15 '20 01:05 sgondala

set CUDA_VISIBLE_DEVICES, it will automatically use all the gpus.

ruotianluo avatar May 15 '20 01:05 ruotianluo

What about the batch size? Should we explicitly increase it (or) would the code automatically take care by multiplying specified batch size by # GPUs?

sgondala avatar May 15 '20 01:05 sgondala

You have to explicityly increase it. The batch size is the total batch size.

ruotianluo avatar May 15 '20 01:05 ruotianluo