Commit 2abd8c1d authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix validation bugs

parent 11cf28f7
...@@ -150,7 +150,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset): ...@@ -150,7 +150,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
name = self.mapping[str(idx)] name = self.mapping[str(idx)]
alignment_dir = os.path.join(self.alignment_dir, name) alignment_dir = os.path.join(self.alignment_dir, name)
if(self.mode == 'train' or self.mode == 'val'): if(self.mode == 'train' or self.mode == 'eval'):
spl = name.rsplit('_', 1) spl = name.rsplit('_', 1)
if(len(spl) == 2): if(len(spl) == 2):
file_id, chain_id = spl file_id, chain_id = spl
...@@ -185,7 +185,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset): ...@@ -185,7 +185,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
return data return data
feats = self.feature_pipeline.process_features( feats = self.feature_pipeline.process_features(
data, self.mode, "unclamped" data, self.mode
) )
return feats return feats
...@@ -426,6 +426,7 @@ class OpenFoldDataModule(pl.LightningDataModule): ...@@ -426,6 +426,7 @@ class OpenFoldDataModule(pl.LightningDataModule):
mapping_path=None, mapping_path=None,
max_template_hits=self.config.eval.max_template_hits, max_template_hits=self.config.eval.max_template_hits,
mode="eval", mode="eval",
_output_raw=True,
) )
else: else:
self.val_dataset = None self.val_dataset = None
......
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