Update train_ms.py

fishaudio · Sep 30, 2023 · dd6aae1 · dd6aae1
1 parent 3ec3a1a
commit dd6aae1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_ms.py b/train_ms.py
@@ -40,6 +40,7 @@
     True
 )  # Not available if torch version is lower than 2.0
 torch.backends.cuda.enable_math_sdp(True)
+torch.multiprocessing.set_start_method('spawn')
 global_step = 0