run_spf_sac_seed.sh

algo=SAC

gpu_id=(7 7 6 6 5 5)
env_id=(HalfCheetah Hopper Walker2d Swimmer Ant Humanoid)
gin_id=(HalfCheetah Hopper Walker2d Swimmer Ant Humanoid)
seed_id=(0 0 0 0 0 0)

for ((i=0;i<${#gpu_id[@]};i++))
do
    CUDA_VISIBLE_DEVICES=${gpu_id[i]} nohup python -u eager_main.py \
                    --policy ${algo} \
                    --env ${env_id[i]}-v2 \
                    --gin ./gins/${gin_id[i]}.gin \
                    --seed ${seed_id[i]} \
                    --fourier_type dtft \
                    --dim_discretize 128 \
                    --use_projection \
                    --projection_dim 512 \
                    --pre_train_step 10000 \
                    --target_update_freq 1000 \
                    --cosine_similarity \
                    --tau 0.01 \
                    --normalizer "batch" \
                    --save_model \
                    --dir-root "./output_${algo}" \
                    --remark "tf-${env_id[i]}, SPF" \
                    > ./my_log/exp_${algo}_${env_id[i]}_spf_s${seed_id[i]}.log 2>&1 &
done