config.GPU = [int(i) para i en config.GPU.split(‘&’) ValueError: literal inválido para int() con base 10: i

Si config.GPU == ‘-1’:
# Horovod
torch.cuda.set_device(hvd.local_rank())
config.GPU = [int(i) for i in range(hvd.size())]
config.g_lr *= hvd.size()
config.d_lr *= hvd.size()

else:
if config.GPU == ‘NO_CUDA’:
config.GPU = ‘-1’
os.environ[“CUDA_VISIBLE_DEVICES”] = config.GPU
config.GPU = [int(i) for i in config.GPU.split(‘,’)]
config.batch_size *= len(config.GPU)
config.g_lr *= len(config.GPU)
config.d_lr *= len(config.GPU)

torch.manual_seed(config.seed)
if torch.cuda.is_available():
torch.cuda.manual_seed_all(config.seed)

config_yaml(config, ‘datasets/{}.yaml’.format(config.dataset_fake))
config = cfg.update_config(config)
if config.mode == ‘train’:
if hvd.rank() == 0:
PRINT(config.log, ‘ ‘.join(sys.argv))
_PRINT(config)
main(config)
config.log.close()

else:
main(config)