From 43157737e5fadd8b3d7da1130e7437a849892f26 Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Mon, 6 Dec 2021 15:16:01 +0100
Subject: [PATCH] Fix Partial Model
---
src/decentralizepy/sharing/PartialModel.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/decentralizepy/sharing/PartialModel.py b/src/decentralizepy/sharing/PartialModel.py
index 7f27d4a..49083aa 100644
--- a/src/decentralizepy/sharing/PartialModel.py
+++ b/src/decentralizepy/sharing/PartialModel.py
@@ -61,7 +61,7 @@ class PartialModel(Sharing):
shared_params["order"] = self.model.state_dict().keys()
shapes = dict()
for k, v in self.model.state_dict().items():
- shapes[k] = v.shape.tolist()
+ shapes[k] = list(v.shape)
shared_params["shapes"] = shapes
shared_params[self.communication_round] = G_topk.tolist()
--
GitLab