diff --git a/train_ms.py b/train_ms.py index 325bee9..7e653df 100644 --- a/train_ms.py +++ b/train_ms.py @@ -37,9 +37,6 @@ from mel_processing import mel_spectrogram_torch, spec_to_mel_torch from text.symbols import symbols torch.backends.cudnn.benchmark = True -torch.backends.cuda.matmul.allow_tf32 = True -torch.backends.cudnn.allow_tf32 = True # If encontered training problem,please try to disable TF32. -torch.set_float32_matmul_precision('medium') torch.backends.cuda.sdp_kernel("flash") torch.backends.cuda.enable_flash_sdp(True) torch.backends.cuda.enable_mem_efficient_sdp(True) # Not avaliable if torch version is lower than 2.0