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

Fix writes of acc, loss; Log which params are shared

parent dd797778
No related branches found
No related tags found
No related merge requests found
...@@ -88,6 +88,7 @@ class Node: ...@@ -88,6 +88,7 @@ class Node:
self.graph = graph self.graph = graph
self.mapping = mapping self.mapping = mapping
self.uid = self.mapping.get_uid(rank, machine_id) self.uid = self.mapping.get_uid(rank, machine_id)
self.log_dir = log_dir
logging.debug("Rank: %d", self.rank) logging.debug("Rank: %d", self.rank)
logging.debug("type(graph): %s", str(type(self.rank))) logging.debug("type(graph): %s", str(type(self.rank)))
...@@ -166,6 +167,7 @@ class Node: ...@@ -166,6 +167,7 @@ class Node:
self.graph, self.graph,
self.model, self.model,
self.dataset, self.dataset,
self.log_dir,
**sharing_params **sharing_params
) )
......
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