Skip to content

Commit

Permalink
change wandb config to not select entity
Browse files Browse the repository at this point in the history
  • Loading branch information
Vindulamj committed Sep 29, 2024
1 parent 9fa05b4 commit 35070da
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 8 deletions.
3 changes: 1 addition & 2 deletions code/ddpg_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')
Expand All @@ -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)

Expand Down
3 changes: 1 addition & 2 deletions code/ppo_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')
Expand All @@ -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)

Expand Down
3 changes: 1 addition & 2 deletions code/synthetic_ddpg_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')
Expand All @@ -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)

Expand Down
3 changes: 1 addition & 2 deletions code/synthetic_ppo_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')
Expand All @@ -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)

Expand Down

0 comments on commit 35070da

Please sign in to comment.