From 23e2b56438ca34aa8509362012b4f4e5339dbe04 Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Thu, 10 Feb 2022 13:52:55 +0100
Subject: [PATCH] Bugfix ChangeAccumulator
---
src/decentralizepy/training/ChangeAccumulator.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/decentralizepy/training/ChangeAccumulator.py b/src/decentralizepy/training/ChangeAccumulator.py
index f485c8b..a463a5b 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()
--
GitLab