mirror of
https://github.com/wassname/curl.git
synced 2026-09-09 11:20:39 +08:00
65 lines
3.0 KiB
Bash
65 lines
3.0 KiB
Bash
#!/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 |