From 1a073a3e8fce5d72ccb7e58ce4cc6ca63f2ea7ea Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Thu, 27 Jan 2022 16:10:30 +0100
Subject: [PATCH] Fixed results dict
---
src/decentralizepy/node/Node.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/src/decentralizepy/node/Node.py b/src/decentralizepy/node/Node.py
index 1115610..c2df24b 100644
--- a/src/decentralizepy/node/Node.py
+++ b/src/decentralizepy/node/Node.py
@@ -305,6 +305,10 @@ class Node:
"test_loss": {},
"test_acc": {},
"total_bytes": {},
+ "total_meta": {},
+ "total_data_per_n": {},
+ "grad_mean": {},
+ "grad_std": {},
}
results_dict["train_loss"][iteration + 1] = loss_after_sharing
--
GitLab