From d6b5e0d27eb5da0b142498b1acb1324cab129647 Mon Sep 17 00:00:00 2001 From: Xiao-Chenguang <973990275@qq.com> Date: Tue, 5 Nov 2024 13:14:57 +0000 Subject: [PATCH] set start method only necessary --- fedmind/server.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/fedmind/server.py b/fedmind/server.py index 53f7e07..356c2bf 100644 --- a/fedmind/server.py +++ b/fedmind/server.py @@ -79,7 +79,8 @@ def __init_mp__(self): """ # Create queues for task distribution and result collection - mp.set_start_method("spawn") + if mp.get_start_method(allow_none=True) is None: + mp.set_start_method("spawn") self._task_queue = mp.Queue() self._result_queue = mp.Queue() self._test_queue = mp.Queue()