From 4d07239005c19797e51d0e50386c087a41395f0d Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Mon, 6 Dec 2021 14:30:47 +0100
Subject: [PATCH] Fix state_dict -> state_dict()
---
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 c15c34a..7f27d4a 100644
--- a/src/decentralizepy/sharing/PartialModel.py
+++ b/src/decentralizepy/sharing/PartialModel.py
@@ -60,7 +60,7 @@ class PartialModel(Sharing):
shared_params = dict()
shared_params["order"] = self.model.state_dict().keys()
shapes = dict()
- for k, v in self.model.state_dict.items():
+ for k, v in self.model.state_dict().items():
shapes[k] = v.shape.tolist()
shared_params["shapes"] = shapes
--
GitLab