diff --git a/climax/core/dataset.py b/climax/core/dataset.py index 7bb306c43eade0bdac0c500f0162ab30c7b46518..d8435cbe5ea02060b32ad50f50a22e4a5d76b5b4 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