From 42a1436a94a7ebd1ff74ea4d0f741f150d790aec Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Fri, 16 Dec 2022 16:17:16 +0100
Subject: [PATCH] fix call to store_snapshot
---
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 b43f1bf..eb23b57 100644
--- a/mu_map/training/lib.py
+++ b/mu_map/training/lib.py
@@ -78,7 +78,7 @@ class AbstractTraining:
self.store_snapshot("val_min")
if epoch % self.snapshot_epoch == 0:
- self._store_snapshot(f"{epoch:0d{len(str(self.epochs))}}")
+ self.store_snapshot(f"{epoch:0d{len(str(self.epochs))}}")
for param in self.training_params:
if param.lr_scheduler is not None:
--
GitLab