modify train.sh

ziyanx02 · Mar 24, 2024 · 006bb7f · 006bb7f
1 parent 45d6616
commit 006bb7f
Showing 1 changed file with 6 additions and 49 deletions.
diff --git a/openrl_ws/train.sh b/openrl_ws/train.sh
@@ -2,18 +2,15 @@ export LD_LIBRARY_PATH=/home/ziyan/anaconda3/envs/mqe/lib
 # task="go1football-defender"
 # task="go1gate"
 task="go1seesaw"
-num_envs=500
-num_steps=40000000
 # task="go1sheep-easy"
-task="go1sheep-hard"
-num_envs=300
-num_steps=24000000
+# task="go1sheep-hard"
 # task="go1pushbox-plane"
 # task="go1pushbox-gate"
-# num_envs=300
-# num_steps=15000000
 # random_seed=0
-device=1
+device=0
+num_envs=500
+num_steps=40000000
+
 # algo="jrpo"
 # cfg=./openrl_ws/cfgs/jrpo.yaml
 algo="ppo"
@@ -33,44 +30,4 @@ python ./openrl_ws/train.py --headless --num_envs $num_envs --train_timesteps $n
     --seed 0 \
     --exp_name test \
     --config $cfg \
-#     --use_wandb
-
-# python ./openrl_ws/train.py --headless --num_envs $num_envs --train_timesteps $num_steps\
-#     --task $task \
-#     --algo $algo \
-#     --sim_device cuda:$device \
-#     --rl_device cuda:$device \
-#     --seed 1 \
-#     --exp_name collect_data \
-#     --config $cfg \
-#     --use_wandb
-
-# python ./openrl_ws/train.py --headless --num_envs $num_envs --train_timesteps $num_steps\
-#     --task $task \
-#     --algo $algo \
-#     --sim_device cuda:$device \
-#     --rl_device cuda:$device \
-#     --seed 2 \
-#     --exp_name collect_data \
-#     --config $cfg \
-#     --use_wandb
-
-# python ./openrl_ws/train.py --headless --num_envs $num_envs --train_timesteps $num_steps\
-#     --task $task \
-#     --algo $algo \
-#     --sim_device cuda:$device \
-#     --rl_device cuda:$device \
-#     --seed 3 \
-#     --exp_name collect_data \
-#     --config $cfg \
-#     --use_wandb
-
-# python ./openrl_ws/train.py --headless --num_envs $num_envs --train_timesteps $num_steps\
-#     --task $task \
-#     --algo $algo \
-#     --sim_device cuda:$device \
-#     --rl_device cuda:$device \
-#     --seed 4 \
-#     --exp_name collect_data \
-#     --config $cfg \
-#     --use_wandb
+    --use_wandb