From 08480c2acc4e209735bdfbc23f725f59f2e2a6d0 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Fri, 16 Dec 2022 16:10:21 +0100
Subject: [PATCH] fix another bug in eval method
---
mu_map/training/lib.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mu_map/training/lib.py b/mu_map/training/lib.py
index ccb2eda..b43f1bf 100644
--- a/mu_map/training/lib.py
+++ b/mu_map/training/lib.py
@@ -112,8 +112,8 @@ class AbstractTraining:
def _eval_epoch(self):
torch.set_grad_enabled(False)
- for model in self.models:
- model.eval()
+ for param in self.training_params:
+ param.model.eval()
loss = 0.0
data_loader = self.data_loaders["validation"]
--
GitLab