From f5ee5eb7835e1281509901db2c00ce7468795baa Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Mon, 18 Oct 2021 16:26:43 +0200
Subject: [PATCH] Learning rate to statefile name.
---
climax/core/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/climax/core/dataset.py b/climax/core/dataset.py
index 7bb306c..d8435cb 100644
--- a/climax/core/dataset.py
+++ b/climax/core/dataset.py
@@ -144,7 +144,7 @@ class EoDataset(torch.utils.data.Dataset):
is not None else state_file)
# add suffix for learning rate values
- state_file = ('_'.join([state_file, 'd{:.0e}'.format(lr)]) if lr
+ state_file = ('_'.join([state_file, 'lr{:.0e}'.format(lr)]) if lr
is not None else state_file)
# add suffix for training with anomalies
--
GitLab