From dd0fa01a77e70d6af6f917a1328ab55b85147f1d Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Thu, 3 Mar 2022 12:11:32 +0100
Subject: [PATCH] Add RandomAlpha, Regular Graph
---
src/decentralizepy/sharing/PartialModel.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/decentralizepy/sharing/PartialModel.py b/src/decentralizepy/sharing/PartialModel.py
index a87b868..107e20d 100644
--- a/src/decentralizepy/sharing/PartialModel.py
+++ b/src/decentralizepy/sharing/PartialModel.py
@@ -159,6 +159,8 @@ class PartialModel(Sharing):
m["params"] = T_topk.numpy()
+ m["send_partial"] = True
+
assert len(m["indices"]) == len(m["params"])
logging.info("Elements sending: {}".format(len(m["indices"])))
@@ -185,7 +187,7 @@ class PartialModel(Sharing):
state_dict of received
"""
- if self.alpha > self.metadata_cap: # Share fully
+ if "send_partial" not in m:
return super().deserialized_model(m)
with torch.no_grad():
--
GitLab