Update train_ms.py

2023-09-06 23:00:14 +08:00
parent bca413b43a
commit 8b61641bb1
1 changed files with 2 additions and 0 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -32,6 +32,8 @@ from losses import generator_loss, discriminator_loss, feature_loss, kl_loss
 from mel_processing import mel_spectrogram_torch, spec_to_mel_torch
 from text.symbols import symbols

+os.environ['NCCL_P2P_DISABLE'] = 1  # Avoid training problem.
+
 torch.backends.cudnn.benchmark = True
 torch.backends.cuda.sdp_kernel("flash")
 torch.backends.cuda.enable_flash_sdp(True)