Fix argparse bug

904de71f · Gustaf Ahdritz · b0313688 · 904de71f
Commit 904de71f authored Aug 01, 2022 by Gustaf Ahdritz
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
@@ -245,7 +245,7 @@ def main(args):
    config = model_config(
        args.config_preset, 
        train=True, 
-        low_prec=(args.precision == "16")
+        low_prec=(str(args.precision) == "16")
    ) 
    model_module = OpenFoldWrapper(config)