Skip to content

GitLab

Commit 0a8ae6a7 authored by Jennifer's avatar Jennifer Committed by Jennifer Wei
Browse files

Updates low_precision check to use current precision settings.

parent cdf60396
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
train_openfold.py
View file @ 0a8ae6a7
...@@ -700,4 +700,5 @@ if __name__ == "__main__": ...@@ -700,4 +700,5 @@ if __name__ == "__main__":
raise ValueError( raise ValueError(
"Choose between loading pretrained Jax-weights and a checkpoint-path") "Choose between loading pretrained Jax-weights and a checkpoint-path")
main(args) main(args)
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