How to allow multi-gpu training in KYS?

[njuxx]

Hi, I added settings.multi_gpu = True and

if settings.multi_gpu:
        net = MultiGPU(net, dim=1)

in ltr/train_settings/kys/kys.py and then ran command "python ltr/run_training.py kys kys", but it still trained only on 1 gpu. So I'd like to know how to train KYS using multiple GPUs? and how to specify the number of gpu to use?

[goutamgmb]

Hi,

The code should use multiple gpus with the modifications you made. Could you try printing torch.cuda.device_count() to check whether multiple gpus are indeed visible to PyTorch?

[DonDominic]

@njuxx I have the same problem as you, have you found a solution?