From 35070da3edd9bd62a5825183d747648b89279f1f Mon Sep 17 00:00:00 2001 From: Vindula Jayawardana Date: Sun, 29 Sep 2024 13:39:53 -0400 Subject: [PATCH] change wandb config to not select entity --- code/ddpg_training.py | 3 +-- code/ppo_training.py | 3 +-- code/synthetic_ddpg_training.py | 3 +-- code/synthetic_ppo_training.py | 3 +-- 4 files changed, 4 insertions(+), 8 deletions(-) diff --git a/code/ddpg_training.py b/code/ddpg_training.py index 2a3c2e5..96f4f2b 100644 --- a/code/ddpg_training.py +++ b/code/ddpg_training.py @@ -37,7 +37,6 @@ parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory') parser.add_argument('--intersection_dir', default='dataset/salt-lake-city', type=str, help='Path to intersection dataset') parser.add_argument('--wandb_project', default='intersectionzoo', type=str, help='Weights and biases project name') -parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name') parser.add_argument('--num_workers', default=10, type=str, help='Number of workers') parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs') parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints') @@ -51,7 +50,7 @@ Path(args.dir).mkdir(parents=True, exist_ok=True) -wandb.init(project=args.wandb_project, entity=args.wandb_entity) +wandb.init(project=args.wandb_project) ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15) diff --git a/code/ppo_training.py b/code/ppo_training.py index 0b0f696..b7e246c 100644 --- a/code/ppo_training.py +++ b/code/ppo_training.py @@ -37,7 +37,6 @@ parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory') parser.add_argument('--intersection_dir', default='dataset/salt-lake-city', type=str, help='Path to intersection dataset') parser.add_argument('--wandb_project', default='intersectionzoo', type=str, help='Weights and biases project name') -parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name') parser.add_argument('--num_workers', default=10, type=str, help='Number of workers') parser.add_argument('--num_gpus', default=1, type=str, help='Number of GPUs') parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints') @@ -51,7 +50,7 @@ Path(args.dir).mkdir(parents=True, exist_ok=True) -wandb.init(project=args.wandb_project, entity=args.wandb_entity) +wandb.init(project=args.wandb_project) ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15) diff --git a/code/synthetic_ddpg_training.py b/code/synthetic_ddpg_training.py index 694189b..23598a8 100644 --- a/code/synthetic_ddpg_training.py +++ b/code/synthetic_ddpg_training.py @@ -36,7 +36,6 @@ parser = argparse.ArgumentParser(description='Model arguments') parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory') parser.add_argument('--wandb_project', default='intersectionzoo-synthetic', type=str, help='Weights and biases project name') -parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name') parser.add_argument('--num_workers', default=10, type=str, help='Number of workers') parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs') parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints') @@ -50,7 +49,7 @@ Path(args.dir).mkdir(parents=True, exist_ok=True) -wandb.init(project=args.wandb_project, entity=args.wandb_entity) +wandb.init(project=args.wandb_project) ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15) diff --git a/code/synthetic_ppo_training.py b/code/synthetic_ppo_training.py index 6b143b0..162ccd7 100644 --- a/code/synthetic_ppo_training.py +++ b/code/synthetic_ppo_training.py @@ -36,7 +36,6 @@ parser = argparse.ArgumentParser(description='Model arguments') parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory') parser.add_argument('--wandb_project', default='intersectionzoo-synthetic', type=str, help='Weights and biases project name') -parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name') parser.add_argument('--num_workers', default=10, type=str, help='Number of workers') parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs') parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints') @@ -50,7 +49,7 @@ Path(args.dir).mkdir(parents=True, exist_ok=True) -wandb.init(project=args.wandb_project, entity=args.wandb_entity) +wandb.init(project=args.wandb_project) ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)