From f353b78a9094e96e7cc5dbafa36b691654248525 Mon Sep 17 00:00:00 2001
From: Jeffrey Wigger <jeffrey.wigger@epfl.ch>
Date: Sat, 19 Mar 2022 19:47:22 +0100
Subject: [PATCH] option for when to evaluate on the trainset fix
---
src/decentralizepy/node/Node.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/decentralizepy/node/Node.py b/src/decentralizepy/node/Node.py
index a543261..74de4e1 100644
--- a/src/decentralizepy/node/Node.py
+++ b/src/decentralizepy/node/Node.py
@@ -379,7 +379,7 @@ class Node:
rounds_to_train_evaluate -= 1
- if rounds_to_test == 0:
+ if rounds_to_train_evaluate == 0:
logging.info("Evaluating on train set.")
rounds_to_train_evaluate = self.train_evaluate_after
loss_after_sharing = self.trainer.eval_loss(self.dataset)
--
GitLab