From 4a1c09bf3149b27dc49d8cb959cd3103f3a163dd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Tue, 5 Sep 2023 00:56:32 +0800 Subject: [PATCH] Update train_ms.py --- train_ms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_ms.py b/train_ms.py index ddf6506..958171e 100644 --- a/train_ms.py +++ b/train_ms.py @@ -38,7 +38,7 @@ from text.symbols import symbols torch.backends.cudnn.benchmark = True torch.backends.cuda.matmul.allow_tf32 = True -torch.backends.cudnn.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)