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