Skip to content
Snippets Groups Projects
Commit f5ee5eb7 authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Learning rate to statefile name.

parent 801e7f3e
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment