From bc15700196b18cdf1f87b7c2c960acde24db66b3 Mon Sep 17 00:00:00 2001 From: Isadora White Date: Sat, 15 Feb 2025 12:02:44 -0800 Subject: [PATCH] fixed wandb logging --- evaluation_script.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/evaluation_script.py b/evaluation_script.py index 1e6c57f..dea7d47 100644 --- a/evaluation_script.py +++ b/evaluation_script.py @@ -114,7 +114,6 @@ def launch_parallel_experiments(task_path, experiments_folder = f"experiments/{exp_name}_{date_time}" # start wandb - subprocess.run(["wandb", "init", "--project", "mindcraft", "--entity", exp_name]) os.makedirs(experiments_folder, exist_ok=True) for i, server in enumerate(servers): launch_server_experiment(task_path, task_ids_split[i], num_exp, server, experiments_folder) @@ -344,9 +343,15 @@ def main(): parser.add_argument('--num_exp', default=1, type=int, help='Number of experiments to run') parser.add_argument('--num_parallel', default=1, type=int, help='Number of parallel servers to run') parser.add_argument('--exp_name', default="exp", help='Name of the experiment') + parser.add_argument('--wandb', action='store_true', help='Whether to use wandb') + parser.add_argument('--wandb-project', default="minecraft_experiements", help='wandb project name') args = parser.parse_args() + if args.wandb: + import wandb + wandb.init(project=args.wandb_project, name=args.exp_name) + # kill all tmux session before starting try: subprocess.run(['tmux', 'kill-server'], check=True)