From dd6aae18a57993f2f9c2a0744417888c57a8e52c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <star_dust_chen@foxmail.com>
Date: Sat, 30 Sep 2023 10:15:44 +0800
Subject: [PATCH] Update train_ms.py

---
 train_ms.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train_ms.py b/train_ms.py
index 8a6fcab24..2aa93852b 100644
--- 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