Commit 6ba0a594 authored by Jennifer's avatar Jennifer
Browse files

adds config recording to wandb

parent d1fe024b
......@@ -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)
......
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