diff --git a/src/decentralizepy/node/Node.py b/src/decentralizepy/node/Node.py index 90140e802d56c0a20323f815f158ad80f8c5acf0..4a6440484a6c925624a1289730be388a745d724e 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 )