From d0694efc86ff0ba3c19bd4ece360eaa3c7ca7ba9 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Fri, 13 Jan 2023 11:08:55 +0100
Subject: [PATCH] fix bug in mu_map trianing lib
---
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 8b27afb..c7b5cc4 100644
--- a/mu_map/training/lib.py
+++ b/mu_map/training/lib.py
@@ -71,7 +71,7 @@ class AbstractTraining:
snapshot_dir: str,
snapshot_epoch: int,
early_stopping: Optional[int] = None,
- logger: Optional[Logger],
+ logger: Optional[Logger] = None,
):
self.epochs = epochs
self.batch_size = batch_size
--
GitLab