From 4fbca222671d18759c5bfe716e045ef7a4a4445e Mon Sep 17 00:00:00 2001 From: x54-729 <17307130121@fudan.edu.cn> Date: Tue, 10 May 2022 07:54:17 +0000 Subject: [PATCH] small --- fastNLP/core/drivers/choose_driver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fastNLP/core/drivers/choose_driver.py b/fastNLP/core/drivers/choose_driver.py index 294bfe28..4be1e502 100644 --- a/fastNLP/core/drivers/choose_driver.py +++ b/fastNLP/core/drivers/choose_driver.py @@ -23,9 +23,9 @@ def choose_driver(model, driver: Union[str, Driver], device: Optional[Union[int, elif driver in {"jittor"}: from fastNLP.core.drivers.jittor_driver.initialize_jittor_driver import initialize_jittor_driver return initialize_jittor_driver(driver, device, model, **kwargs) - elif driver in {"paddle", "fleet"}: + elif driver in {"paddle"}: from fastNLP.core.drivers.paddle_driver.initialize_paddle_driver import initialize_paddle_driver return initialize_paddle_driver(driver, device, model, **kwargs) else: raise ValueError("Parameter `driver` can only be one of these values: ['torch', 'fairscale', " - "'jittor', 'paddle', 'fleet'].") \ No newline at end of file + "'jittor', 'paddle'].") \ No newline at end of file