state mask

This commit is contained in:
MishaLaskin
2020-03-10 19:21:55 -07:00
parent 108972dece
commit d767f8ec4c
4 changed files with 48 additions and 3 deletions
+1 -1
View File
@@ -4,6 +4,6 @@ CUDA_VISIBLE_DEVICES=5 python train.py \
--encoder_type pixel \
--action_repeat 8 \
--save_tb --pre_transform_image_size 100 --image_size 84 \
--work_dir ./tmp/cartpole \
--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