Browse Source

small

tags/v1.0.0alpha
x54-729 3 years ago
parent
commit
262bc1a82e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py

+ 1
- 1
fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py View File

@@ -28,7 +28,7 @@ def initialize_paddle_driver(driver: str, device: Optional[Union[str, int, List[
"""
if is_in_paddle_launch_dist():
if device is not None:
logger.warning("Parameter `device` would be ignored when you are using `paddle.distributed.launch` to pull "
logger.warning_once("Parameter `device` would be ignored when you are using `paddle.distributed.launch` to pull "
"up your script. And we will directly get the local device via "
"and `os.environ['CUDA_VISIBLE_DEVICES']``.")
device = [int(g) for g in os.environ["CUDA_VISIBLE_DEVICES"].split(",")]


Loading…
Cancel
Save