diff --git a/asym_sac/train.py b/asym_sac/train.py index 9e6a132..9428de5 100644 --- a/asym_sac/train.py +++ b/asym_sac/train.py @@ -365,7 +365,7 @@ def main(args): # WandB if args.wandb: wandb_logger = wandb.init( - project="cto-rl-manipulation", + project="dmfd", config=args.__dict__, sync_tensorboard=True, # auto-upload tensorboard metrics name=args.name) diff --git a/awac/AWAC/awac.py b/awac/AWAC/awac.py index 7b16113..29a8119 100644 --- a/awac/AWAC/awac.py +++ b/awac/AWAC/awac.py @@ -272,7 +272,7 @@ def __init__(self, args, env_kwargs): if args['wandb']: self.wandb_run = wandb.init( - project="cto-rl-manipulation", + project="dmfd", config=args, name=args['folder_name'], ) diff --git a/col/train.py b/col/train.py index 8c4d97c..44dd375 100644 --- a/col/train.py +++ b/col/train.py @@ -349,7 +349,7 @@ def main(args): # WandB if args.wandb: wandb_logger = wandb.init( - project="cto-rl-manipulation", + project="dmfd", config=args.__dict__, sync_tensorboard=True, # auto-upload tensorboard metrics name=args.name) diff --git a/curl/train.py b/curl/train.py index 0095856..b868563 100644 --- a/curl/train.py +++ b/curl/train.py @@ -332,7 +332,7 @@ def main(args): # WandB if args.wandb: wandb_logger = wandb.init( - project="cto-rl-manipulation", + project="dmfd", config=args.__dict__, sync_tensorboard=True, # auto-upload tensorboard metrics name=args.name) diff --git a/drq/train.py b/drq/train.py index 5b6f639..ccb8eff 100644 --- a/drq/train.py +++ b/drq/train.py @@ -143,7 +143,7 @@ def __init__(self, cfg): cfg_dict = cfg.__dict__ cfg_dict['run_name'] = cfg.name self.wandb_run = wandb.init( - project="cto-rl-manipulation", + project="dmfd", config=cfg_dict, sync_tensorboard=True, # auto-upload sb3's tensorboard metrics name=cfg_dict['run_name'], diff --git a/experiments/run_bc.py b/experiments/run_bc.py index f5edee3..a120ff6 100644 --- a/experiments/run_bc.py +++ b/experiments/run_bc.py @@ -499,7 +499,7 @@ def main_training(): if args.wandb: wandb.init( - project="cto-rl-manipulation", + project="dmfd", config={k: v for k, v in args.__dict__.items()}, name=args.folder_name, ) diff --git a/sb3/train.py b/sb3/train.py index bde1e22..f55401c 100644 --- a/sb3/train.py +++ b/sb3/train.py @@ -197,7 +197,7 @@ def run_task(args, env_kwargs): tb_callback = TensorboardCallback(verbose=0) if args.wandb: run = wandb.init( - project="cto-rl-manipulation", + project="dmfd", config=args.__dict__, sync_tensorboard=True, # auto-upload sb3's tensorboard metrics name=args.folder_name,