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

adds config recording to wandb

parent d1fe024b
...@@ -383,6 +383,7 @@ def main(args): ...@@ -383,6 +383,7 @@ def main(args):
save_dir=args.output_dir, save_dir=args.output_dir,
id=args.wandb_id, id=args.wandb_id,
project=args.wandb_project, project=args.wandb_project,
config=config.to_dict(),
**{"entity": args.wandb_entity} **{"entity": args.wandb_entity}
) )
loggers.append(wdb_logger) 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