From 327827e06c425e71485c375d93727c6196971e66 Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Mon, 6 Dec 2021 15:42:40 +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 49083aa..69875ad 100644
--- a/src/decentralizepy/sharing/PartialModel.py
+++ b/src/decentralizepy/sharing/PartialModel.py
@@ -58,7 +58,7 @@ class PartialModel(Sharing):
shared_params = json.load(inf)
else:
shared_params = dict()
- shared_params["order"] = self.model.state_dict().keys()
+ shared_params["order"] = list(self.model.state_dict().keys())
shapes = dict()
for k, v in self.model.state_dict().items():
shapes[k] = list(v.shape)
--
GitLab