tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update train_ms.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-20 15:30:26 +08:00
committed by GitHub
parent 28e450fd5b
commit 8f0c3340bc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_ms.py
View File

@@ -49,7 +49,7 @@ def main():
n_gpus = torch.cuda.device_count() n_gpus = torch.cuda.device_count()
os.environ['MASTER_ADDR'] = 'localhost' os.environ['MASTER_ADDR'] = 'localhost'
os.environ['MASTER_PORT'] = '8000' os.environ['MASTER_PORT'] = '65280'
hps = utils.get_hparams() hps = utils.get_hparams()
mp.spawn(run, nprocs=n_gpus, args=(n_gpus, hps,)) mp.spawn(run, nprocs=n_gpus, args=(n_gpus, hps,))