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

Fixed results dict

parent 713c0ff3
Branches
Tags
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment