From 35e6983fecf87ec652ae61ff86e10bbc5ff2d3cf Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Mon, 6 Dec 2021 14:21:36 +0100
Subject: [PATCH] Fix writes of acc, loss; Log which params are shared
---
src/decentralizepy/node/Node.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/decentralizepy/node/Node.py b/src/decentralizepy/node/Node.py
index 90140e8..4a64404 100644
--- a/src/decentralizepy/node/Node.py
+++ b/src/decentralizepy/node/Node.py
@@ -88,6 +88,7 @@ class Node:
self.graph = graph
self.mapping = mapping
self.uid = self.mapping.get_uid(rank, machine_id)
+ self.log_dir = log_dir
logging.debug("Rank: %d", self.rank)
logging.debug("type(graph): %s", str(type(self.rank)))
@@ -166,6 +167,7 @@ class Node:
self.graph,
self.model,
self.dataset,
+ self.log_dir,
**sharing_params
)
--
GitLab