From 5aaa9722f5c9f7327e74b1453629fa1a4cd1b351 Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Wed, 23 Feb 2022 21:51:15 +0100
Subject: [PATCH] Add ManualAdapt
---
src/decentralizepy/sharing/ManualAdapt.py | 14 ++++++++++++--
1 file changed, 12 insertions(+), 2 deletions(-)
diff --git a/src/decentralizepy/sharing/ManualAdapt.py b/src/decentralizepy/sharing/ManualAdapt.py
index 8c653a6..dcb94cf 100644
--- a/src/decentralizepy/sharing/ManualAdapt.py
+++ b/src/decentralizepy/sharing/ManualAdapt.py
@@ -19,8 +19,8 @@ class ManualAdapt(PartialModel):
model,
dataset,
log_dir,
- change_alpha: list,
- change_rounds: list,
+ change_alpha,
+ change_rounds,
dict_ordered=True,
save_shared=False,
metadata_cap=1.0,
@@ -58,6 +58,16 @@ class ManualAdapt(PartialModel):
Share full model when self.alpha > metadata_cap
"""
+ assert change_alpha != ""
+ assert change_alpha != None
+ assert change_rounds != ""
+ assert change_rounds != None
+
+ if type(change_alpha) == str:
+ change_alpha = eval(change_alpha)
+ if type(change_rounds) == str:
+ change_rounds = eval(change_rounds)
+
super().__init__(
rank,
machine_id,
--
GitLab