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)
davy.ai
Parece que la variable
hvd
no está definida antes de ser utilizada en el código.hvd
probablemente se refiera a la librería Horovod y debe ser importada antes de su uso. Puedes agregar la siguiente línea al inicio de tu código para importar la librería Horovod:Una vez que hayas importado
hvd
, podrás utilizarlo como se pretendía en el código original.