From d300a3da8f46b9f6bc93c357da7a03fdd5357197 Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Thu, 27 Jan 2022 16:16:50 +0100
Subject: [PATCH] Bug fix
---
src/decentralizepy/node/Node.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/decentralizepy/node/Node.py b/src/decentralizepy/node/Node.py
index c2df24b..ef66d17 100644
--- a/src/decentralizepy/node/Node.py
+++ b/src/decentralizepy/node/Node.py
@@ -314,15 +314,15 @@ class Node:
results_dict["train_loss"][iteration + 1] = loss_after_sharing
results_dict["total_bytes"][iteration + 1] = self.communication.total_bytes
- if self.sharing.total_meta:
+ if hasattr(self.sharing, "total_meta"):
results_dict["total_meta"][iteration + 1] = self.sharing.total_meta
- if self.sharing.total_data:
+ if hasattr(self.sharing, "total_data"):
results_dict["total_data_per_n"][
iteration + 1
] = self.sharing.total_data
- if self.sharing.mean:
+ if hasattr(self.sharing, "mean"):
results_dict["grad_mean"][iteration + 1] = self.sharing.mean
- if self.sharing.std:
+ if hasattr(self.sharing, "std"):
results_dict["grad_std"][iteration + 1] = self.sharing.std
self.save_plot(
--
GitLab