Browse Source

删除debug信息

dev0.8.0
x54-729 2 years ago
parent
commit
5ebc8d1be4
1 changed files with 0 additions and 1 deletions
  1. +0
    -1
      fastNLP/core/drivers/torch_driver/initialize_torch_driver.py

+ 0
- 1
fastNLP/core/drivers/torch_driver/initialize_torch_driver.py View File

@@ -63,7 +63,6 @@ def initialize_torch_driver(driver: str, device: Optional[Union[str, "torch.devi
raise ValueError("Parameter `device` can only be '-1' when it is smaller than 0.") raise ValueError("Parameter `device` can only be '-1' when it is smaller than 0.")
device = [torch.device(f"cuda:{w}") for w in range(_could_use_device_num)] device = [torch.device(f"cuda:{w}") for w in range(_could_use_device_num)]
elif device >= _could_use_device_num: elif device >= _could_use_device_num:
print(device, _could_use_device_num)
raise ValueError("The gpu device that parameter `device` specifies is not existed.") raise ValueError("The gpu device that parameter `device` specifies is not existed.")
else: else:
device = torch.device(f"cuda:{device}") device = torch.device(f"cuda:{device}")


Loading…
Cancel
Save