From 2da7d4f2634108a8d644e8186b6bbed0f5ec92d9 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Fri, 16 Dec 2022 16:08:04 +0100
Subject: [PATCH] eval_epoch no longer requires the phase argument
---
mu_map/training/lib.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mu_map/training/lib.py b/mu_map/training/lib.py
index dc92337..ccb2eda 100644
--- a/mu_map/training/lib.py
+++ b/mu_map/training/lib.py
@@ -110,7 +110,7 @@ class AbstractTraining:
param.optimizer.step()
return loss / len(data_loader)
- def _eval_epoch(self, phase: str):
+ def _eval_epoch(self):
torch.set_grad_enabled(False)
for model in self.models:
model.eval()
--
GitLab