From 8b61641bb1109632716c75b47aa935ce1fb55c59 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Wed, 6 Sep 2023 23:00:14 +0800 Subject: [PATCH] Update train_ms.py --- train_ms.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train_ms.py b/train_ms.py index a8d1eb8..7619237 100644 --- 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)