From e42be5d2a435ff7c22586c81b9f00b863ea6abdc Mon Sep 17 00:00:00 2001 From: x54-729 <17307130121@fudan.edu.cn> Date: Sun, 8 May 2022 13:08:23 +0000 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=20initialize=5Fpaddle=5Fdriv?= =?UTF-8?q?er=20=E4=B8=AD=E7=9A=84=20warning=20=E4=B8=BA=20info?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../drivers/paddle_driver/initialize_paddle_driver.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py b/fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py index 66da8cf1..f07cd47e 100644 --- a/fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py +++ b/fastNLP/core/drivers/paddle_driver/initialize_paddle_driver.py @@ -69,17 +69,17 @@ def initialize_paddle_driver(driver: str, device: Optional[Union[str, int, List[ if not isinstance(device, List): return PaddleSingleDriver(model, device, **kwargs) else: - logger.rank_zero_warning("Notice you are using `paddle` driver but your chosen `device` are multi gpus, we will use" - "`Fleetriver` by default. But if you mean using `PaddleFleetDriver`, you should choose parameter" - "`driver` as `PaddleFleetDriver`.") + logger.info("Notice you are using `paddle` driver but your chosen `device` are multi gpus, we will use" + "`PaddleFleetDriver` by default. But if you mean using `PaddleFleetDriver`, you should choose parameter" + "`driver` as `fleet`.") return PaddleFleetDriver(model, device, **kwargs) elif driver == "fleet": if not isinstance(device, List): if device == "cpu": raise ValueError("You are using `fleet` driver, but your chosen `device` is 'cpu'.") - logger.rank_zero_warning("Notice you are using `fleet` driver, but your chosen `device` is only one gpu, we will" + logger.info("Notice you are using `fleet` driver, but your chosen `device` is only one gpu, we will" "still use `PaddleFleetDriver` for you, but if you mean using `PaddleSingleDriver`, you should " - "choose `paddle` driver.") + "`driver` as `paddle`.") return PaddleFleetDriver(model, [device], **kwargs) else: return PaddleFleetDriver(model, device, **kwargs)