diff --git a/scripts/run_cheetah.sh b/scripts/run_cheetah.sh deleted file mode 100644 index 11762a3..0000000 --- a/scripts/run_cheetah.sh +++ /dev/null @@ -1,65 +0,0 @@ -#!/bin/bash -# 256 -# 2e4 lr 2x -# 5e4 lr 2x -# 1e3 lr 2x - -CUDA_VISIBLE_DEVICES=1 python train_cpc.py \ - --domain_name cheetah \ - --task_name run --dmc2gym \ - --encoder_type pixel \ - --decoder_type identity \ - --action_repeat 4 --batch_size 256 \ - --save_tb --work_dir ./tmp/icml/feb3cheetah/curl_cheetah_b256_84_lr2e4_a \ - --agent sac_cpc --frame_stack 3 --pre_transform_image_size 100 --image_size 84 \ - --seed 23 --encoder_lr 2e-4 --critic_lr 2e-4 --actor_lr 2e-4 --eval_freq 20000 --batch_size 128 --num_train_steps 3000000 & - -CUDA_VISIBLE_DEVICES=2 python train_cpc.py \ - --domain_name cheetah \ - --task_name run --dmc2gym \ - --encoder_type pixel \ - --decoder_type identity \ - --action_repeat 4 --batch_size 256 \ - --save_tb --work_dir ./tmp/icml/feb3cheetah/curl_cheetah_b256_84_lr2e4_b \ - --agent sac_cpc --frame_stack 3 --pre_transform_image_size 100 --image_size 84 \ - --seed -1 --encoder_lr 2e-4 --critic_lr 2e-4 --actor_lr 2e-4 --eval_freq 20000 --batch_size 128 --num_train_steps 3000000 & - -CUDA_VISIBLE_DEVICES=3 python train_cpc.py \ - --domain_name cheetah \ - --task_name run --dmc2gym \ - --encoder_type pixel \ - --decoder_type identity \ - --action_repeat 4 --batch_size 256 \ - --save_tb --work_dir ./tmp/icml/feb3cheetah/curl_cheetah_b256_84_lr5e4_a \ - --agent sac_cpc --frame_stack 3 --pre_transform_image_size 100 --image_size 84 \ - --seed 23 --encoder_lr 5e-4 --critic_lr 5e-4 --actor_lr 5e-4 --eval_freq 20000 --batch_size 128 --num_train_steps 3000000 & - -CUDA_VISIBLE_DEVICES=4 python train_cpc.py \ - --domain_name cheetah \ - --task_name run --dmc2gym \ - --encoder_type pixel \ - --decoder_type identity \ - --action_repeat 4 --batch_size 256 \ - --save_tb --work_dir ./tmp/icml/feb3cheetah/curl_cheetah_b256_84_lr5e4_b \ - --agent sac_cpc --frame_stack 3 --pre_transform_image_size 100 --image_size 84 \ - --seed -1 --encoder_lr 5e-4 --critic_lr 5e-4 --actor_lr 5e-4 --eval_freq 20000 --batch_size 128 --num_train_steps 3000000 & - -CUDA_VISIBLE_DEVICES=5 python train_cpc.py \ - --domain_name cheetah \ - --task_name run --dmc2gym \ - --encoder_type pixel \ - --decoder_type identity \ - --action_repeat 4 --batch_size 256 \ - --save_tb --work_dir ./tmp/icml/feb3cheetah/curl_cheetah_b256_84_lr1e3_a \ - --agent sac_cpc --frame_stack 3 --pre_transform_image_size 100 --image_size 84 \ - --seed 23 --encoder_lr 1e-3 --critic_lr 1e-3 --actor_lr 1e-3 --eval_freq 20000 --batch_size 128 --num_train_steps 3000000 & - -CUDA_VISIBLE_DEVICES=6 python train_cpc.py \ - --domain_name cheetah \ - --task_name run --dmc2gym \ - --encoder_type pixel \ - --decoder_type identity \ - --action_repeat 4 --batch_size 256 \ - --save_tb --work_dir ./tmp/icml/feb3cheetah/curl_cheetah_b256_84_lr1e3_b \ - --agent sac_cpc --frame_stack 3 --pre_transform_image_size 100 --image_size 84 \ - --seed -1 --encoder_lr 1e-3 --critic_lr 1e-3 --actor_lr 1e-3 --eval_freq 20000 --batch_size 128 --num_train_steps 3000000 \ No newline at end of file diff --git a/scripts/run_multiple.sh b/scripts/run_multiple.sh deleted file mode 100644 index 2872872..0000000 --- a/scripts/run_multiple.sh +++ /dev/null @@ -1,17 +0,0 @@ -#!/bin/bash - -gpuList=(1 2 3 4 5 6 7) -( -for i in ${gpuList[@]}; do - CUDA_VISIBLE_DEVICES=$i python train.py \ - --domain_name ball_in_cup \ - --task_name catch \ - --encoder_type pixel \ - --action_repeat 4 \ - --save_tb --pre_transform_image_size 100 --image_size 84 \ - --work_dir ./tmp/curl/ball_in_cup \ - --agent curl_sac --frame_stack 3 \ - --seed -1 --critic_lr 1e-3 --actor_lr 1e-3 --eval_freq 20000 --batch_size 128 --num_train_steps 1000000 & -done -) -echo "Launched CURL script" \ No newline at end of file