Fix type bug in training script

c392b21c · Gustaf Ahdritz · GitHub · 3718d10f · c392b21c
Unverified Commit c392b21c authored Jan 27, 2022 by Gustaf Ahdritz Committed by GitHub Jan 27, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

train_openfold.py train_openfold.py +1 -1

No files found.
--- a/train_openfold.py
+++ b/train_openfold.py
@@ -112,7 +112,7 @@ def main(args):
    config = model_config(
        "model_1", 
        train=True, 
-        low_prec=(args.precision == 16)
+        low_prec=(args.precision == "16")
    ) 
    
    model_module = OpenFoldWrapper(config)