有bug

[ZouRuia]

tools.py里 我用cpu进行训练，n_gpu=‘’， def prepare_device(use_gpu): """ setup GPU device if available, move model into configured device # 如果n_gpu_use为数字，则使用range生成list # 如果输入的是一个list，则默认使用list[0]作为controller Example: use_gpu = '' : cpu use_gpu = '0': cuda:0 use_gpu = '0,1' : cuda:0 and cuda:1 """ # use_gpu = '' n_gpu_use = [int(x) for x in use_gpu.split(",")] if not use_gpu: device_type = 'cpu' else: device_type = f"cuda:{n_gpu_use[0]}" n_gpu = torch.cuda.device_count() if len(n_gpu_use) > 0 and n_gpu == 0: logger.warning("Warning: There's no GPU available on this machine, training will be performed on CPU.") device_type = 'cpu' if len(n_gpu_use) > n_gpu: msg = f"Warning: The number of GPU's configured to use is {n_gpu}, but only {n_gpu} are available on this machine." logger.warning(msg) n_gpu_use = range(n_gpu) device = torch.device(device_type) list_ids = n_gpu_use return device, list_ids 这里n_gpu_use = [int(x) for x in use_gpu.split(",")]会报错，原因是use_gpu是空的字符串，不能int.