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
267b1bfd
Commit
267b1bfd
authored
Oct 20, 2021
by
Gustaf Ahdritz
Browse files
Add warning to run script
parent
a3a27143
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
3 deletions
+9
-3
run_pretrained_openfold.py
run_pretrained_openfold.py
+9
-3
No files found.
run_pretrained_openfold.py
View file @
267b1bfd
...
@@ -182,7 +182,7 @@ if __name__ == "__main__":
...
@@ -182,7 +182,7 @@ if __name__ == "__main__":
)
)
parser
.
add_argument
(
parser
.
add_argument
(
"--cpus"
,
type
=
int
,
default
=
4
,
"--cpus"
,
type
=
int
,
default
=
4
,
help
=
"""Number of CPUs
to use
to run alignment tools"""
help
=
"""Number of CPUs
with which
to run alignment tools"""
)
)
parser
.
add_argument
(
parser
.
add_argument
(
'--preset'
,
type
=
str
,
default
=
'full_dbs'
,
'--preset'
,
type
=
str
,
default
=
'full_dbs'
,
...
@@ -200,6 +200,12 @@ if __name__ == "__main__":
...
@@ -200,6 +200,12 @@ if __name__ == "__main__":
"params_"
+
args
.
model_name
+
".npz"
"params_"
+
args
.
model_name
+
".npz"
)
)
if
(
args
.
model_device
==
"cpu"
and
torch
.
cuda
.
is_available
()):
logging
.
warning
(
"""The model is being run on CPU. Consider specifying
--model_device for better performance"""
)
if
(
args
.
bfd_database_path
is
None
and
if
(
args
.
bfd_database_path
is
None
and
args
.
small_bfd_database_path
is
None
):
args
.
small_bfd_database_path
is
None
):
raise
ValueError
(
raise
ValueError
(
...
...
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