Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenFold
Commits
2abd8c1d
Commit
2abd8c1d
authored
Dec 09, 2021
by
Gustaf Ahdritz
Browse files
Fix validation bugs
parent
11cf28f7
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
2 deletions
+3
-2
openfold/data/data_modules.py
openfold/data/data_modules.py
+3
-2
No files found.
openfold/data/data_modules.py
View file @
2abd8c1d
...
...
@@ -150,7 +150,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
name
=
self
.
mapping
[
str
(
idx
)]
alignment_dir
=
os
.
path
.
join
(
self
.
alignment_dir
,
name
)
if
(
self
.
mode
==
'train'
or
self
.
mode
==
'val'
):
if
(
self
.
mode
==
'train'
or
self
.
mode
==
'
e
val'
):
spl
=
name
.
rsplit
(
'_'
,
1
)
if
(
len
(
spl
)
==
2
):
file_id
,
chain_id
=
spl
...
...
@@ -185,7 +185,7 @@ class OpenFoldSingleDataset(torch.utils.data.Dataset):
return
data
feats
=
self
.
feature_pipeline
.
process_features
(
data
,
self
.
mode
,
"unclamped"
data
,
self
.
mode
)
return
feats
...
...
@@ -426,6 +426,7 @@ class OpenFoldDataModule(pl.LightningDataModule):
mapping_path
=
None
,
max_template_hits
=
self
.
config
.
eval
.
max_template_hits
,
mode
=
"eval"
,
_output_raw
=
True
,
)
else
:
self
.
val_dataset
=
None
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment