diff --git a/experiments/scripts/d4rl/door/launch_calql.sh b/experiments/scripts/d4rl/door/launch_calql.sh index 932275f..0c4c698 100644 --- a/experiments/scripts/d4rl/door/launch_calql.sh +++ b/experiments/scripts/d4rl/door/launch_calql.sh @@ -25,7 +25,6 @@ do python main/offline_to_online_rl/finetune_gym.py \ --agent calql \ --config experiments/configs/train_config.py:adroit_cql \ - --project q2rl \ --group door_calql \ --warmup_steps 0 \ --num_offline_steps 20_000 \ diff --git a/experiments/scripts/d4rl/door/launch_cql.sh b/experiments/scripts/d4rl/door/launch_cql.sh index 35fb6df..ef3e108 100644 --- a/experiments/scripts/d4rl/door/launch_cql.sh +++ b/experiments/scripts/d4rl/door/launch_cql.sh @@ -25,7 +25,6 @@ do python main/offline_to_online_rl/finetune_gym.py \ --agent cql \ --config experiments/configs/train_config.py:adroit_cql \ - --project q2rl \ --group door_cql \ --warmup_steps 0 \ --num_offline_steps 20_000 \ diff --git a/experiments/scripts/d4rl/door/launch_ibrl.sh b/experiments/scripts/d4rl/door/launch_ibrl.sh index 06cfb08..7ff8b08 100644 --- a/experiments/scripts/d4rl/door/launch_ibrl.sh +++ b/experiments/scripts/d4rl/door/launch_ibrl.sh @@ -26,7 +26,6 @@ do python3 main/ibrl/ibrl_gym.py \ --agent ibrl_state \ --config experiments/configs/train_config.py:adroit_wsrl \ - --project q2rl \ --group door_ibrl \ --reward_scale 10.0 \ --reward_bias 5.0 \ diff --git a/experiments/scripts/d4rl/door/launch_q2rl.sh b/experiments/scripts/d4rl/door/launch_q2rl.sh index e1ed7aa..7c668c5 100644 --- a/experiments/scripts/d4rl/door/launch_q2rl.sh +++ b/experiments/scripts/d4rl/door/launch_q2rl.sh @@ -26,7 +26,6 @@ do python3 main/q2rl/q2rl_gym.py \ --agent q2rl \ --config experiments/configs/train_config.py:adroit_wsrl \ - --project q2rl \ --group door_q2rl \ --reward_scale 10.0 \ --reward_bias 5.0 \ diff --git a/experiments/scripts/d4rl/door/launch_wsrl.sh b/experiments/scripts/d4rl/door/launch_wsrl.sh index bdde818..9ee50bc 100644 --- a/experiments/scripts/d4rl/door/launch_wsrl.sh +++ b/experiments/scripts/d4rl/door/launch_wsrl.sh @@ -25,7 +25,6 @@ do python3 main/offline_to_online_rl/finetune_gym.py \ --agent sac \ --config experiments/configs/train_config.py:adroit_wsrl \ - --project q2rl \ --group door_wsrl \ --num_offline_steps 20_000 \ --resume_path $PROJECT_ROOT/data/policies/rl_offline/door/calql/checkpoint_20000 \ diff --git a/experiments/scripts/d4rl/kitchen/launch_ibrl.sh b/experiments/scripts/d4rl/kitchen/launch_ibrl.sh index 0c03144..03dd2fa 100644 --- a/experiments/scripts/d4rl/kitchen/launch_ibrl.sh +++ b/experiments/scripts/d4rl/kitchen/launch_ibrl.sh @@ -25,7 +25,6 @@ do python3 main/ibrl/ibrl_gym.py \ --agent ibrl_state \ --config experiments/configs/train_config.py:kitchen_wsrl \ - --project q2rl \ --group kitchen_ibrl \ --reward_scale 1.0 \ --reward_bias -4.0 \ diff --git a/experiments/scripts/d4rl/kitchen/launch_q2rl.sh b/experiments/scripts/d4rl/kitchen/launch_q2rl.sh index 8b0494c..2fc3a5e 100644 --- a/experiments/scripts/d4rl/kitchen/launch_q2rl.sh +++ b/experiments/scripts/d4rl/kitchen/launch_q2rl.sh @@ -25,7 +25,6 @@ do python3 main/q2rl/q2rl_gym.py \ --agent q2rl \ --config experiments/configs/train_config.py:kitchen_wsrl \ - --project q2rl \ --group kitchen_q2rl \ --reward_scale 1.0 \ --reward_bias -4.0 \ diff --git a/experiments/scripts/d4rl/kitchen/launch_wsrl.sh b/experiments/scripts/d4rl/kitchen/launch_wsrl.sh index 65978a3..683c8a1 100644 --- a/experiments/scripts/d4rl/kitchen/launch_wsrl.sh +++ b/experiments/scripts/d4rl/kitchen/launch_wsrl.sh @@ -25,7 +25,6 @@ do python3 main/offline_to_online_rl/finetune_gym.py \ --agent sac \ --config experiments/configs/train_config.py:kitchen_wsrl \ - --project q2rl \ --group kitchen_wsrl \ --num_offline_steps 250_000 \ --resume_path $PROJECT_ROOT/data/policies/rl_offline/kitchen/calql/checkpoint_250000 \ diff --git a/experiments/scripts/d4rl/pen/launch_q2rl.sh b/experiments/scripts/d4rl/pen/launch_q2rl.sh index 206f596..b7d7d3d 100644 --- a/experiments/scripts/d4rl/pen/launch_q2rl.sh +++ b/experiments/scripts/d4rl/pen/launch_q2rl.sh @@ -26,7 +26,6 @@ do python3 main/q2rl/q2rl_gym.py \ --agent q2rl \ --config experiments/configs/train_config.py:adroit_wsrl \ - --project q2rl \ --group pen_q2rl \ --reward_scale 10.0 \ --reward_bias 5.0 \ diff --git a/experiments/scripts/d4rl/pen/launch_wsrl.sh b/experiments/scripts/d4rl/pen/launch_wsrl.sh index 327cbe0..63b5bf7 100644 --- a/experiments/scripts/d4rl/pen/launch_wsrl.sh +++ b/experiments/scripts/d4rl/pen/launch_wsrl.sh @@ -26,7 +26,6 @@ do --agent sac \ --group pen_wsrl \ --config experiments/configs/train_config.py:adroit_wsrl \ - --project q2rl \ --num_offline_steps 20_000 \ --resume_path $PROJECT_ROOT/data/policies/rl_offline/pen/calql/checkpoint_20000 \ --reward_scale 10.0 \ diff --git a/experiments/scripts/robosuite-img/can-img/launch_ibrl_with_data.sh b/experiments/scripts/robosuite-img/can-img/launch_ibrl_with_data.sh index 6dbe042..fca0054 100644 --- a/experiments/scripts/robosuite-img/can-img/launch_ibrl_with_data.sh +++ b/experiments/scripts/robosuite-img/can-img/launch_ibrl_with_data.sh @@ -17,7 +17,7 @@ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia seed=(0 10 20 30 40) -for i in {0..1} +for i in {0..4} do python3 main/ibrl/ibrl_robosuite_image.py \ --agent=ibrl_image \ diff --git a/experiments/scripts/robosuite-img/can-img/launch_ibrl_without_data.sh b/experiments/scripts/robosuite-img/can-img/launch_ibrl_without_data.sh index 8ddf5f4..4170485 100644 --- a/experiments/scripts/robosuite-img/can-img/launch_ibrl_without_data.sh +++ b/experiments/scripts/robosuite-img/can-img/launch_ibrl_without_data.sh @@ -17,7 +17,7 @@ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia seed=(0 10 20 30 40) -for i in {0..1} +for i in {0..4} do python3 main/ibrl/ibrl_robosuite_image.py \ --agent=ibrl_image \ diff --git a/experiments/scripts/robosuite-img/can-img/launch_q2rl_with_data.sh b/experiments/scripts/robosuite-img/can-img/launch_q2rl_with_data.sh index 68fbdea..9c65dd4 100644 --- a/experiments/scripts/robosuite-img/can-img/launch_q2rl_with_data.sh +++ b/experiments/scripts/robosuite-img/can-img/launch_q2rl_with_data.sh @@ -16,8 +16,8 @@ export MUJOCO_GL=egl export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia -seed=(0 10 20) -for i in {0..1} +seed=(0 10 20 30 40) +for i in {0..4} do python3 main/q2rl/q2rl_robosuite_image.py \ --agent=q2rl_rs_image \ diff --git a/experiments/scripts/robosuite-img/can-img/launch_q2rl_without_data.sh b/experiments/scripts/robosuite-img/can-img/launch_q2rl_without_data.sh index b37d06f..495d630 100644 --- a/experiments/scripts/robosuite-img/can-img/launch_q2rl_without_data.sh +++ b/experiments/scripts/robosuite-img/can-img/launch_q2rl_without_data.sh @@ -16,8 +16,8 @@ export MUJOCO_GL=egl export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia -seed=(0 10 20) -for i in {0..1} +seed=(0 10 20 30 40) +for i in {0..4} do python3 main/q2rl/q2rl_robosuite_image.py \ --agent=q2rl_rs_image \ diff --git a/experiments/scripts/robosuite-img/lift-img/launch_ibrl_with_data.sh b/experiments/scripts/robosuite-img/lift-img/launch_ibrl_with_data.sh index 2818df3..875c961 100644 --- a/experiments/scripts/robosuite-img/lift-img/launch_ibrl_with_data.sh +++ b/experiments/scripts/robosuite-img/lift-img/launch_ibrl_with_data.sh @@ -17,7 +17,7 @@ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia seed=(0 10 20 30 40) -for i in {0..1} +for i in {0..4} do python3 main/ibrl/ibrl_robosuite_image.py \ --agent=ibrl_image \ diff --git a/experiments/scripts/robosuite-img/lift-img/launch_ibrl_without_data.sh b/experiments/scripts/robosuite-img/lift-img/launch_ibrl_without_data.sh index 1a04363..4fba43e 100644 --- a/experiments/scripts/robosuite-img/lift-img/launch_ibrl_without_data.sh +++ b/experiments/scripts/robosuite-img/lift-img/launch_ibrl_without_data.sh @@ -17,7 +17,7 @@ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia seed=(0 10 20 30 40) -for i in {0..1} +for i in {0..4} do python3 main/ibrl/ibrl_robosuite_image.py \ --agent=ibrl_image \ diff --git a/experiments/scripts/robosuite-img/lift-img/launch_q2rl_with_data.sh b/experiments/scripts/robosuite-img/lift-img/launch_q2rl_with_data.sh index 4f12e62..2ac25c4 100644 --- a/experiments/scripts/robosuite-img/lift-img/launch_q2rl_with_data.sh +++ b/experiments/scripts/robosuite-img/lift-img/launch_q2rl_with_data.sh @@ -16,8 +16,8 @@ export MUJOCO_GL=egl export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia -seed=(0 10 20) -for i in {0..1} +seed=(0 10 20 30 40) +for i in {0..4} do python3 main/q2rl/q2rl_robosuite_image.py \ --agent=q2rl_rs_image \ diff --git a/experiments/scripts/robosuite-img/lift-img/launch_q2rl_without_data.sh b/experiments/scripts/robosuite-img/lift-img/launch_q2rl_without_data.sh index 48a957e..17e3591 100644 --- a/experiments/scripts/robosuite-img/lift-img/launch_q2rl_without_data.sh +++ b/experiments/scripts/robosuite-img/lift-img/launch_q2rl_without_data.sh @@ -16,8 +16,8 @@ export MUJOCO_GL=egl export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia -seed=(0 10 20) -for i in {0..1} +seed=(0 10 20 30 40) +for i in {0..4} do python3 main/q2rl/q2rl_robosuite_image.py \ --agent=q2rl_rs_image \ diff --git a/experiments/scripts/robosuite-state/can-state/launch_ibrl_with_data.sh b/experiments/scripts/robosuite-state/can-state/launch_ibrl_with_data.sh index 31f8fc8..84f6364 100644 --- a/experiments/scripts/robosuite-state/can-state/launch_ibrl_with_data.sh +++ b/experiments/scripts/robosuite-state/can-state/launch_ibrl_with_data.sh @@ -22,7 +22,6 @@ do python3 main/ibrl/ibrl_robosuite.py \ --agent ibrl_state \ --config experiments/configs/train_config.py:robosuite_ibrl \ - --project q2rl \ --group can_state_ibrl_with_data \ --get_demo_buffer=True \ --data_filter_key=train \ diff --git a/experiments/scripts/robosuite-state/can-state/launch_ibrl_without_data.sh b/experiments/scripts/robosuite-state/can-state/launch_ibrl_without_data.sh index f024ebc..cf77b00 100644 --- a/experiments/scripts/robosuite-state/can-state/launch_ibrl_without_data.sh +++ b/experiments/scripts/robosuite-state/can-state/launch_ibrl_without_data.sh @@ -23,7 +23,6 @@ do python3 main/ibrl/ibrl_robosuite.py \ --agent ibrl_state \ --config experiments/configs/train_config.py:robosuite_ibrl \ - --project q2rl \ --group can_state_ibrl_without_data \ --get_demo_buffer=False \ --reward_scale 5.0 \ diff --git a/experiments/scripts/robosuite-state/lift-state/launch_ibrl_with_data.sh b/experiments/scripts/robosuite-state/lift-state/launch_ibrl_with_data.sh index e732128..0366d47 100644 --- a/experiments/scripts/robosuite-state/lift-state/launch_ibrl_with_data.sh +++ b/experiments/scripts/robosuite-state/lift-state/launch_ibrl_with_data.sh @@ -22,7 +22,6 @@ do python3 main/ibrl/ibrl_robosuite.py \ --agent ibrl_state \ --config experiments/configs/train_config.py:robosuite_ibrl \ - --project q2rl \ --group lift_state_ibrl_with_data \ --get_demo_buffer=True \ --data_filter_key=20_percent_train \ diff --git a/experiments/scripts/robosuite-state/lift-state/launch_q2rl_with_data.sh b/experiments/scripts/robosuite-state/lift-state/launch_q2rl_with_data.sh index 71d52fd..5c0a624 100644 --- a/experiments/scripts/robosuite-state/lift-state/launch_q2rl_with_data.sh +++ b/experiments/scripts/robosuite-state/lift-state/launch_q2rl_with_data.sh @@ -18,8 +18,8 @@ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia export DATASET_PATH=$(python -c "from huggingface_hub import snapshot_download; print(snapshot_download('theaiinstitute/q2rl_robomimic_datasets', repo_type='dataset'))") -seed=(0 10 20) -for i in {0..2} +seed=(0 10 20 30 40) +for i in {0..4} do python3 main/q2rl/q2rl_robosuite.py \ --agent=q2rl_rs \ diff --git a/experiments/scripts/robosuite-state/lift-state/launch_q2rl_without_data.sh b/experiments/scripts/robosuite-state/lift-state/launch_q2rl_without_data.sh index 8da13a6..258c60a 100644 --- a/experiments/scripts/robosuite-state/lift-state/launch_q2rl_without_data.sh +++ b/experiments/scripts/robosuite-state/lift-state/launch_q2rl_without_data.sh @@ -16,8 +16,8 @@ export MUJOCO_GL=egl export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco210/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia -seed=(0 10 20) -for i in {0..2} +seed=(0 10 20 30 40) +for i in {0..4} do python3 main/q2rl/q2rl_robosuite.py \ --agent=q2rl_rs \ diff --git a/experiments/scripts/robosuite-state/square-state/launch_ibrl_with_data.sh b/experiments/scripts/robosuite-state/square-state/launch_ibrl_with_data.sh index f7daa5e..7fea6b4 100644 --- a/experiments/scripts/robosuite-state/square-state/launch_ibrl_with_data.sh +++ b/experiments/scripts/robosuite-state/square-state/launch_ibrl_with_data.sh @@ -22,7 +22,6 @@ do python3 main/ibrl/ibrl_robosuite.py \ --agent ibrl_state \ --config experiments/configs/train_config.py:robosuite_ibrl \ - --project q2rl \ --group square_state_ibrl_with_data \ --get_demo_buffer=True \ --data_filter_key=train \ diff --git a/experiments/scripts/robosuite-state/square-state/launch_ibrl_without_data.sh b/experiments/scripts/robosuite-state/square-state/launch_ibrl_without_data.sh index 4b5e14e..6b929f3 100644 --- a/experiments/scripts/robosuite-state/square-state/launch_ibrl_without_data.sh +++ b/experiments/scripts/robosuite-state/square-state/launch_ibrl_without_data.sh @@ -23,7 +23,6 @@ do python3 main/ibrl/ibrl_robosuite.py \ --agent ibrl_state \ --config experiments/configs/train_config.py:robosuite_ibrl \ - --project q2rl \ --group square_state_ibrl_without_data \ --get_demo_buffer=False \ --reward_scale 5.0 \ diff --git a/experiments/scripts/robosuite-state/square-state/launch_q2rl_with_data.sh b/experiments/scripts/robosuite-state/square-state/launch_q2rl_with_data.sh index 5b48555..520ab3a 100644 --- a/experiments/scripts/robosuite-state/square-state/launch_q2rl_with_data.sh +++ b/experiments/scripts/robosuite-state/square-state/launch_q2rl_with_data.sh @@ -22,7 +22,6 @@ do python3 main/q2rl/q2rl_robosuite.py \ --agent q2rl_rs \ --config experiments/configs/train_config.py:robosuite_q2rl \ - --project q2rl \ --group square_state_q2rl_with_data \ --get_demo_buffer=True \ --demo_path data/datasets/square/ph/low_dim_v141.hdf5 \ diff --git a/experiments/scripts/robosuite-state/square-state/launch_q2rl_without_data.sh b/experiments/scripts/robosuite-state/square-state/launch_q2rl_without_data.sh index 393ebad..b278f16 100644 --- a/experiments/scripts/robosuite-state/square-state/launch_q2rl_without_data.sh +++ b/experiments/scripts/robosuite-state/square-state/launch_q2rl_without_data.sh @@ -23,7 +23,6 @@ do python3 main/q2rl/q2rl_robosuite.py \ --agent q2rl_rs \ --config experiments/configs/train_config.py:robosuite_q2rl \ - --project q2rl \ --group square_state_q2rl_without_data \ --get_demo_buffer=False \ --reward_scale 5.0 \ diff --git a/main/ibrl/ibrl_gym.py b/main/ibrl/ibrl_gym.py index 971d906..06a0ed1 100644 --- a/main/ibrl/ibrl_gym.py +++ b/main/ibrl/ibrl_gym.py @@ -77,7 +77,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") flags.DEFINE_bool("get_new_rollouts", False, "get new rollouts") @@ -100,7 +100,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.group}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", } diff --git a/main/ibrl/ibrl_robosuite.py b/main/ibrl/ibrl_robosuite.py index 308fc71..76f096f 100644 --- a/main/ibrl/ibrl_robosuite.py +++ b/main/ibrl/ibrl_robosuite.py @@ -87,7 +87,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") flags.DEFINE_string("demo_path", None, "path to load the demo replay buffer from") @@ -113,7 +113,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.group}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", } diff --git a/main/ibrl/ibrl_robosuite_image.py b/main/ibrl/ibrl_robosuite_image.py index 6998092..007e4b6 100644 --- a/main/ibrl/ibrl_robosuite_image.py +++ b/main/ibrl/ibrl_robosuite_image.py @@ -76,7 +76,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") flags.DEFINE_string("replay_path", None, "path to load the replay buffer from") @@ -106,7 +106,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.group}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", } diff --git a/main/offline_to_online_rl/finetune_gym.py b/main/offline_to_online_rl/finetune_gym.py index 4ab6c35..6b10538 100644 --- a/main/offline_to_online_rl/finetune_gym.py +++ b/main/offline_to_online_rl/finetune_gym.py @@ -86,7 +86,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") @@ -124,7 +124,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.exp_name}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", } diff --git a/main/offline_to_online_rl/finetune_robosuite.py b/main/offline_to_online_rl/finetune_robosuite.py index 5389fe8..fd6c716 100644 --- a/main/offline_to_online_rl/finetune_robosuite.py +++ b/main/offline_to_online_rl/finetune_robosuite.py @@ -99,7 +99,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") flags.DEFINE_bool("get_demo_buffer", False, "Load demo trajs into demo buffer") @@ -134,7 +134,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.exp_name}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", } diff --git a/main/offline_to_online_rl/finetune_robosuite_image.py b/main/offline_to_online_rl/finetune_robosuite_image.py index a3de1a6..3dfaac5 100644 --- a/main/offline_to_online_rl/finetune_robosuite_image.py +++ b/main/offline_to_online_rl/finetune_robosuite_image.py @@ -99,7 +99,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") flags.DEFINE_bool("get_demo_buffer", False, "Load demo trajs into demo buffer") @@ -136,7 +136,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.exp_name}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", } diff --git a/main/q2rl/q2rl_gym.py b/main/q2rl/q2rl_gym.py index 849425f..9803388 100644 --- a/main/q2rl/q2rl_gym.py +++ b/main/q2rl/q2rl_gym.py @@ -72,7 +72,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") @@ -98,7 +98,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.group}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", } diff --git a/main/q2rl/q2rl_robosuite.py b/main/q2rl/q2rl_robosuite.py index 0cf81c3..5d33a97 100644 --- a/main/q2rl/q2rl_robosuite.py +++ b/main/q2rl/q2rl_robosuite.py @@ -78,7 +78,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") flags.DEFINE_bool("get_new_rollouts", False, "get new rollouts") @@ -106,7 +106,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.group}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", } diff --git a/main/q2rl/q2rl_robosuite_image.py b/main/q2rl/q2rl_robosuite_image.py index 8abb317..aaaafb6 100644 --- a/main/q2rl/q2rl_robosuite_image.py +++ b/main/q2rl/q2rl_robosuite_image.py @@ -74,7 +74,7 @@ # wandb flags.DEFINE_string("exp_name", "", "Experiment name for wandb logging") -flags.DEFINE_string("project", None, "Wandb project folder") +flags.DEFINE_string("project", "q2rl", "Wandb project folder") flags.DEFINE_string("group", None, "Wandb group of the experiment") flags.DEFINE_bool("debug", False, "If true, no logging to wandb") flags.DEFINE_bool("get_new_rollouts", False, "get new rollouts") @@ -108,7 +108,7 @@ def main(_): wandb_config = WandBLogger.get_default_config() wandb_config.update( { - "project": "q2rl" or FLAGS.project, + "project": FLAGS.project, "group": FLAGS.group, "exp_descriptor": f"{FLAGS.group}_{FLAGS.env}_{FLAGS.agent}_seed{FLAGS.seed}", }