diff --git a/src/decentralizepy/node/Node.py b/src/decentralizepy/node/Node.py index c2df24b56bb5bcfd48a8a269b1da2c4b634deec6..ef66d1752dae14a341b8c04042ee42961544f91a 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(