use a deepcopy to store gm_param

Xiao-Chenguang · Oct 28, 2024 · fb4127f · fb4127f
1 parent fd0b9d7
commit fb4127f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fedmind/server.py b/fedmind/server.py
@@ -42,7 +42,7 @@ def __init__(
         self._criterion = criterion
         self.args = args
 
-        self._gm_params = self._model.state_dict(destination=StateDict())
+        self._gm_params = self._model.state_dict(destination=StateDict()) * 1
         optim: dict = self.args.OPTIM
         if optim["NAME"] == "SGD":
             self._optimizer = SGD(self._model.parameters(), lr=optim["LR"])