diff --git a/fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py b/fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py index 98655757..2cba6388 100644 --- a/fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py +++ b/fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py @@ -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(",")]