Files
2020-03-10 20:30:00 -07:00

9 lines
388 B
Bash
Executable File

CUDA_VISIBLE_DEVICES=5 python train.py \
--domain_name cartpole \
--task_name swingup \
--encoder_type identity \
--action_repeat 8 \
--save_tb --pre_transform_image_size 100 --image_size 84 \
--work_dir ./tmp/test \
--agent curl_sac --frame_stack 3 \
--seed -1 --critic_lr 1e-3 --actor_lr 1e-3 --eval_freq 10000 --batch_size 128 --num_train_steps 1000000