diff --git a/src/decentralizepy/training/ChangeAccumulator.py b/src/decentralizepy/training/ChangeAccumulator.py index f485c8b4a06c72c4f3cb23672b9ca141d7fc8c04..a463a5b696585e20fd22c12c56bcf81fb23db841 100644 --- a/src/decentralizepy/training/ChangeAccumulator.py +++ b/src/decentralizepy/training/ChangeAccumulator.py @@ -101,8 +101,8 @@ class ChangeAccumulator(Training): output_dict = dict() output_dict["order"] = list(self.model.state_dict().keys()) shapes = dict() - for k, v in self.model.state_dict().items(): - shapes[k] = list(v.shape) + for k, v1 in self.model.state_dict().items(): + shapes[k] = list(v1.shape) output_dict["shapes"] = shapes output_dict[self.communication_round] = v.tolist()