Skip to content
Snippets Groups Projects
Commit d300a3da authored by Rishi Sharma's avatar Rishi Sharma
Browse files

Bug fix

parent 1a073a3e
No related branches found
No related tags found
No related merge requests found
......@@ -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(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment