44 lines
1.1 KiB
Bash
44 lines
1.1 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
DOMAIN=carla096
|
||
|
|
TASK=highway
|
||
|
|
AGENT=deepmdp
|
||
|
|
SEED=5
|
||
|
|
DECODER_TYPE=identity
|
||
|
|
TRANSITION_MODEL=deterministic
|
||
|
|
|
||
|
|
SAVEDIR=./save
|
||
|
|
#SAVEDIR=/checkpoint/${USER}/pixel-pets/carla/${AGENT}_${TRANSITION_MODEL}_currrew_${DECODER_TYPE}/seed_${SEED}
|
||
|
|
mkdir -p ${SAVEDIR}
|
||
|
|
|
||
|
|
CUDA_VISIBLE_DEVICES=1 python train.py \
|
||
|
|
--domain_name ${DOMAIN} \
|
||
|
|
--task_name ${TASK} \
|
||
|
|
--agent ${AGENT} \
|
||
|
|
--init_steps 100 \
|
||
|
|
--num_train_steps 100000 \
|
||
|
|
--encoder_type pixelCarla096 \
|
||
|
|
--decoder_type ${DECODER_TYPE} \
|
||
|
|
--resource_files 'distractors/*.mp4' \
|
||
|
|
--action_repeat 4 \
|
||
|
|
--critic_tau 0.01 \
|
||
|
|
--encoder_tau 0.05 \
|
||
|
|
--encoder_stride 2 \
|
||
|
|
--decoder_weight_lambda 0.0000001 \
|
||
|
|
--hidden_dim 1024 \
|
||
|
|
--replay_buffer_capacity 100000 \
|
||
|
|
--total_frames 10000 \
|
||
|
|
--num_layers 4 \
|
||
|
|
--num_filters 32 \
|
||
|
|
--batch_size 128 \
|
||
|
|
--init_temperature 0.1 \
|
||
|
|
--alpha_lr 1e-4 \
|
||
|
|
--alpha_beta 0.5 \
|
||
|
|
--work_dir ${SAVEDIR} \
|
||
|
|
--transition_model_type ${TRANSITION_MODEL} \
|
||
|
|
--seed ${SEED} $@ \
|
||
|
|
--frame_stack 3 \
|
||
|
|
--image_size 84 \
|
||
|
|
--save_model >> ${SAVEDIR}/output.txt \
|
||
|
|
# --render
|