From 8f0c3340bc2e88c73309bde881262e5fda9911c8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com> Date: Sun, 20 Aug 2023 15:30:26 +0800 Subject: [PATCH] Update train_ms.py --- train_ms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_ms.py b/train_ms.py index 28e602a..7080e91 100644 --- a/train_ms.py +++ b/train_ms.py @@ -49,7 +49,7 @@ def main(): n_gpus = torch.cuda.device_count() os.environ['MASTER_ADDR'] = 'localhost' - os.environ['MASTER_PORT'] = '8000' + os.environ['MASTER_PORT'] = '65280' hps = utils.get_hparams() mp.spawn(run, nprocs=n_gpus, args=(n_gpus, hps,))