adds config recording to wandb

aqlaboratory · Mar 13, 2024 · 6ba0a59 · 6ba0a59
1 parent d1fe024
commit 6ba0a59
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_openfold.py b/train_openfold.py
@@ -383,6 +383,7 @@ def main(args):
             save_dir=args.output_dir,
             id=args.wandb_id,
             project=args.wandb_project,
+            config=config.to_dict(),
             **{"entity": args.wandb_entity}
         )
         loggers.append(wdb_logger)