Hi @HyeyeonKoo
for multi-GPU training you need to launch the script using torch.distributed.launch, have a look at this section of example docs
Hi @HyeyeonKoo
for multi-GPU training you need to launch the script using torch.distributed.launch, have a look at this section of example docs