Browse Source

修改 initialize_paddle_driver 中的 warning 为 info

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

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

@@ -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)

Loading…
Cancel
Save