wassname/curl
1
0
Fork 0
mirror of https://github.com/wassname/curl.git synced 2026-09-10 12:00:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

scripts

Browse Source
This commit is contained in:
MishaLaskin
2020-02-19 20:52:43 -08:00
parent 0f2f957711
commit 06115f1a75
2 changed files with 0 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/run_cheetah.sh
-65
View File
@@ -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
scripts/run_multiple.sh
-17
View File
@@ -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"