37 lines
844 B
Bash
37 lines
844 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
DOMAIN=carla098
|
||
|
|
TASK=highway
|
||
|
|
|
||
|
|
SAVEDIR=./save
|
||
|
|
mkdir -p ${SAVEDIR}
|
||
|
|
|
||
|
|
CUDA_VISIBLE_DEVICES=0 python train.py \
|
||
|
|
--domain_name ${DOMAIN} \
|
||
|
|
--task_name ${TASK} \
|
||
|
|
--agent 'deepmdp' \
|
||
|
|
--init_steps 1000 \
|
||
|
|
--num_train_steps 1000000 \
|
||
|
|
--encoder_type pixelCarla098 \
|
||
|
|
--decoder_type pixel \
|
||
|
|
--img_source video \
|
||
|
|
--resource_files 'distractors/*.mp4' \
|
||
|
|
--action_repeat 4 \
|
||
|
|
--critic_tau 0.01 \
|
||
|
|
--encoder_tau 0.05 \
|
||
|
|
--decoder_weight_lambda 0.0000001 \
|
||
|
|
--hidden_dim 1024 \
|
||
|
|
--total_frames 10000 \
|
||
|
|
--num_filters 32 \
|
||
|
|
--batch_size 128 \
|
||
|
|
--init_temperature 0.1 \
|
||
|
|
--alpha_lr 1e-4 \
|
||
|
|
--alpha_beta 0.5 \
|
||
|
|
--work_dir ${SAVEDIR}/${DOMAIN}_${TASK} \
|
||
|
|
--seed 1 $@ \
|
||
|
|
--frame_stack 3 \
|
||
|
|
--image_size 84 \
|
||
|
|
--eval_freq 10000 \
|
||
|
|
--num_eval_episodes 25 \
|
||
|
|
--render
|