diff --git a/src/decentralizepy/node/Node.py b/src/decentralizepy/node/Node.py index a5432618b9e93c58f22aa3059ac7ee6070373760..74de4e1376bb1bae0b713e7c1b7d439316e7facb 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)