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