Skip to content
Snippets Groups Projects
Commit f353b78a authored by Jeffrey Wigger's avatar Jeffrey Wigger
Browse files

option for when to evaluate on the trainset fix

parent b6096c2e
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
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