[bug] fix bug

SMILELab-FL · Apr 4, 2022 · 2a24ecc · 2a24ecc
1 parent e2d3cdf
commit 2a24ecc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/fedlab/core/communicator/package.py b/fedlab/core/communicator/package.py
@@ -92,7 +92,8 @@ def append_tensor(self, tensor):
 
         shape = list(tensor.shape)
         slice = [tensor.numel(), len(shape)] + shape
-
+
+        tensor = tensor.view(-1)
         if self.content is None:
             self.content = deepcopy(tensor)
             self.dtype = tensor.dtype