Skip to content

Commit 35070da

Browse files
committed
change wandb config to not select entity
1 parent 9fa05b4 commit 35070da

File tree

4 files changed

+4
-8
lines changed

4 files changed

+4
-8
lines changed

code/ddpg_training.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,6 @@
3737
parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
3838
parser.add_argument('--intersection_dir', default='dataset/salt-lake-city', type=str, help='Path to intersection dataset')
3939
parser.add_argument('--wandb_project', default='intersectionzoo', type=str, help='Weights and biases project name')
40-
parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name')
4140
parser.add_argument('--num_workers', default=10, type=str, help='Number of workers')
4241
parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs')
4342
parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints')
@@ -51,7 +50,7 @@
5150

5251
Path(args.dir).mkdir(parents=True, exist_ok=True)
5352

54-
wandb.init(project=args.wandb_project, entity=args.wandb_entity)
53+
wandb.init(project=args.wandb_project)
5554

5655
ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)
5756

code/ppo_training.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,6 @@
3737
parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
3838
parser.add_argument('--intersection_dir', default='dataset/salt-lake-city', type=str, help='Path to intersection dataset')
3939
parser.add_argument('--wandb_project', default='intersectionzoo', type=str, help='Weights and biases project name')
40-
parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name')
4140
parser.add_argument('--num_workers', default=10, type=str, help='Number of workers')
4241
parser.add_argument('--num_gpus', default=1, type=str, help='Number of GPUs')
4342
parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints')
@@ -51,7 +50,7 @@
5150

5251
Path(args.dir).mkdir(parents=True, exist_ok=True)
5352

54-
wandb.init(project=args.wandb_project, entity=args.wandb_entity)
53+
wandb.init(project=args.wandb_project)
5554

5655
ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)
5756

code/synthetic_ddpg_training.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@
3636
parser = argparse.ArgumentParser(description='Model arguments')
3737
parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
3838
parser.add_argument('--wandb_project', default='intersectionzoo-synthetic', type=str, help='Weights and biases project name')
39-
parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name')
4039
parser.add_argument('--num_workers', default=10, type=str, help='Number of workers')
4140
parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs')
4241
parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints')
@@ -50,7 +49,7 @@
5049

5150
Path(args.dir).mkdir(parents=True, exist_ok=True)
5251

53-
wandb.init(project=args.wandb_project, entity=args.wandb_entity)
52+
wandb.init(project=args.wandb_project)
5453

5554
ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)
5655

code/synthetic_ppo_training.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@
3636
parser = argparse.ArgumentParser(description='Model arguments')
3737
parser.add_argument('--dir', default='wd/new_exp', type=str, help='Result directory')
3838
parser.add_argument('--wandb_project', default='intersectionzoo-synthetic', type=str, help='Weights and biases project name')
39-
parser.add_argument('--wandb_entity', default='run_1', type=str, help='Weights and biases entity name')
4039
parser.add_argument('--num_workers', default=10, type=str, help='Number of workers')
4140
parser.add_argument('--num_gpus', default=0, type=str, help='Number of GPUs')
4241
parser.add_argument('--save_frequency', default=5, type=str, help='Frequency of saving checkpoints')
@@ -50,7 +49,7 @@
5049

5150
Path(args.dir).mkdir(parents=True, exist_ok=True)
5251

53-
wandb.init(project=args.wandb_project, entity=args.wandb_entity)
52+
wandb.init(project=args.wandb_project)
5453

5554
ray.init(ignore_reinit_error=True, num_cpus=args.num_workers + 15)
5655

0 commit comments

Comments
 (0)