source /home/kun/anaconda3/bin/activate torch

current_time=$(date +"%Y%m%d%H%M%S")

desired_directory="/mnt/storage/result"

if [ ! -d "$desired_directory" ]; then

mkdir -p "$desired_directory"

cd "$desired_directory"

param1_values=(6) # num of agents

param2_values=(8 )

param3_values=(16)

param4_values=('True') # token query, work for trans only

param5_values=('True' ) # reduce state or original state, original state is not well maintained

param6_values=(1) # num of encoder

param7_values=(2) # num of decoder

param8_values=( 20) #reset level; level 2: random 1 piece; level 3: random 2 pieces; level 13: random 3 pieces;

param9_values=('fc10_3e') # network model

param10_values=(True) # whether to share layers

param11_values=(1e-5) # beta base

param12_values=(1e-5) # beta range

param13_values=(2) # num of future corridors in state, at least 1

param14_values=(0.3 ) # acceleration max

param15_values=( 128 ) # capability of corridor indexing

param16_values=( 'True' ) # rotate cylinder for simple

param17_values=(2) # state choice

param18_values=( 'all') # state choice

param19_values=(4) # state choice

max_concurrent=100

concurrent_processes=0

num_executions=1

num_task_gpu0=0

num_task_gpu1=0

for i in $(seq $num_executions); do

for param1 in "${param1_values[@]}"; do

for param2 in "${param2_values[@]}"; do

for param3 in "${param3_values[@]}"; do

for param4 in "${param4_values[@]}"; do

for param5 in "${param5_values[@]}"; do

for param6 in "${param6_values[@]}"; do

for param7 in "${param7_values[@]}"; do

for param8 in "${param8_values[@]}"; do

for param9 in "${param9_values[@]}"; do

for param10 in "${param10_values[@]}"; do

for param11 in "${param11_values[@]}"; do

for param12 in "${param12_values[@]}"; do

for param13 in "${param13_values[@]}"; do

for param14 in "${param14_values[@]}"; do

for param15 in "${param15_values[@]}"; do

for param16 in "${param16_values[@]}"; do

for param17 in "${param17_values[@]}"; do

for param18 in "${param18_values[@]}"; do

for param19 in "${param19_values[@]}"; do

if awk -v p11="$param11" -v p12="$param12" 'BEGIN{ exit !(p11 == 1e-5 && p12 == 1.1) }'; then

continue # Skip this combination

fi

gpu_index=1 # Set default GPU indexn

# Construct experiment name

exp_name="new_net:${param6}_${param17}_${param18}_${param19}_${param11}_${param12}_model_${param9}_horizon${param2}_level${param8}_capacity${param1}"

echo $PATH

# Run the Python script with parameters

CUDA_VISIBLE_DEVICES=$gpu_index python /home/kun/PycharmProjects/air-corridor_ncfo/rl_multi_3d_trans/main.py \

--seed 55 \

--curriculum True \

--num_agents ${param1} \

--multiply_horrizion ${param2} \

--multiply_batch ${param3} \

--level ${param8} \

--net_model ${param9} \

--K_epochs ${param19} \

--Max_train_steps 3e7 \

--exp-name ${exp_name} &

# Manage concurrent processes

concurrent_processes=$((concurrent_processes + 1))

# Limit the number of concurrent processes

if [ "$concurrent_processes" -ge "$max_concurrent" ]; then

wait

concurrent_processes=0

fi

done

done

done

done

done

done

done

done

done

done

done

done

done

done

done

done

done

done

done