From 24ca2851bbca2a3273dfa3d4e2c1956af1afd339 Mon Sep 17 00:00:00 2001 From: x54-729 <17307130121@fudan.edu.cn> Date: Mon, 17 Oct 2022 16:16:27 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9ddp=20python=20-m=20a.b.c=20?= =?UTF-8?q?=E5=90=AF=E5=8A=A8=E6=97=B6=E7=9A=84=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/core/drivers/paddle_driver/fleet_launcher.py | 2 +- fastNLP/core/drivers/torch_driver/ddp.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/fastNLP/core/drivers/paddle_driver/fleet_launcher.py b/fastNLP/core/drivers/paddle_driver/fleet_launcher.py index f6d99333..4a42dcff 100644 --- a/fastNLP/core/drivers/paddle_driver/fleet_launcher.py +++ b/fastNLP/core/drivers/paddle_driver/fleet_launcher.py @@ -83,7 +83,7 @@ class FleetLauncher: # use the same python interpreter and actually running command = [sys.executable] + command else: # Script called as `python -m a.b.c` - command = [sys.executable, "-m", __main__.__spec__._name] + sys.argv[1:] + command = [sys.executable, "-m", __main__.__spec__.name] + sys.argv[1:] current_env = copy.copy(self.global_envs) for idx, t in enumerate(self.sub_trainers): diff --git a/fastNLP/core/drivers/torch_driver/ddp.py b/fastNLP/core/drivers/torch_driver/ddp.py index 2bf38a0d..d9f08a2d 100644 --- a/fastNLP/core/drivers/torch_driver/ddp.py +++ b/fastNLP/core/drivers/torch_driver/ddp.py @@ -425,7 +425,7 @@ class TorchDDPDriver(TorchDriver): command = [sys.executable] + command # Script called as `python -m a.b.c` else: - command = [sys.executable, "-m", __main__.__spec__._name] + sys.argv[1:] + command = [sys.executable, "-m", __main__.__spec__.name] + sys.argv[1:] os.environ['MASTER_ADDR'] = self.master_address os.environ['MASTER_PORT'] = self.master_port