diff --git a/train_ms.py b/train_ms.py index b1cc3a4..9f8ab5f 100644 --- a/train_ms.py +++ b/train_ms.py @@ -36,6 +36,7 @@ from style_bert_vits2.models.models import ( from style_bert_vits2.nlp.symbols import SYMBOLS from style_bert_vits2.utils.stdout_wrapper import SAFE_STDOUT + torch.backends.cuda.matmul.allow_tf32 = True torch.backends.cudnn.allow_tf32 = ( True # If encontered training problem,please try to disable TF32.