Change config preset in training script

60d3b198 · Gustaf Ahdritz · 16d75057 · 60d3b198
Commit 60d3b198 authored Feb 08, 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
@@ -127,7 +127,7 @@ def main(args):
        seed_everything(args.seed) 

    config = model_config(
-        "model_1", 
+        "initial_training", 
        train=True, 
        low_prec=(args.precision == "16")
    )