Merge pull request #67 from Xiao-Chenguang/flex-alg

Use inplace add to avoid broke the reference to model parameters

fedmind/algs/fedavg.py

@@ -31,6 +31,6 @@ def _aggregate_updates(self, updates: list[dict]) -> dict:
         """
         agg_update = sum([update["model_update"] for update in updates]) / len(updates)
         agg_loss = sum([update["train_loss"] for update in updates]) / len(updates)
-        self._gm_params += agg_update
+        self._gm_params.add_(agg_update)
         self.logger.info(f"Train loss: {agg_loss:.4f}")
         return {"train_loss": agg_loss}