Update train_ms.py

2023-08-20 15:30:26 +08:00
parent 28e450fd5b
commit 8f0c3340bc
1 changed files with 1 additions and 1 deletions
--- 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,))