Commit 1f709b0d authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Update name of validation mode

parent 0febc3e5
......@@ -90,7 +90,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
self.mode = mode
self._output_raw = _output_raw
valid_modes = ["train", "val", "predict"]
valid_modes = ["train", "eval", "predict"]
if(mode not in valid_modes):
raise ValueError(f'mode must be one of {valid_modes}')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment