I’m using this open-source code: https://github.com/sweetcocoa/ddsp-pytorch/blob/master/train/train.py 4
Except I don’t want to use just one gpu, so I modified up until line 78 in the file linked above to be:
config = setup(default_config="…/configs/violin.yaml")
print(OmegaConf.create(config.dict).pretty())
set_seeds(config.seed)
Trainer.set_experiment_name(config.experiment_name)
list_of_gpus = [int(s) for s in config.gpu_list.split(’,’)]
torch.cuda.set_device(0)
net = AutoEncoder(config)
net.to(0)
net = torch.nn.DataParallel(net, device_ids = list_of_gpus)