run_meta.sh

#!/bin/bash --login
#SBATCH -N 1
#SBATCH --partition=batch
#SBATCH -J tp
#SBATCH -o tp.%J.out
#SBATCH -e tp.%J.err
#SBATCH --array=1-12
#SBATCH --time=00:30:00
#SBATCH --mem=128G
#SBATCH --gres=gpu

# load the module
conda activate /ibex/scratch/zhanc0c/projects/fed_wireless/FedTP/bin/env

alg_values=( 'meta' 'avg')
type_values=('avg' 'att' 'dist' )
file_values=( 'milano' 'trentino')
adapt_lr_values=(0.001)
lr_values=(0.01)
shot_values=(5)
adapt_steps_values=(5)
out_dim_values=(1)
granularity_values=(6)
fo_values=( no-fo )

trial=${SLURM_ARRAY_TASK_ID}
alg=${alg_values[$((trial % ${#alg_values[@]}))]}
trial=$((trial / ${#alg_values[@]}))
type=${type_values[$((trial % ${#type_values[@]}))]}
trial=$((trial / ${#type_values[@]}))
file=${file_values[$((trial % ${#file_values[@]}))]}

trial=$((trial / ${#file_values[@]}))
adapt_lr=${adapt_lr_values[$((trial % ${#adapt_lr_values[@]}))]}
trial=$((trial / ${#adapt_lr_values[@]}))
lr=${lr_values[$((trial % ${#lr_values[@]}))]}
trial=$((trial / ${#lr_values[@]}))
shots=${shot_values[$((trial % ${#shot_values[@]}))]}
trial=$((trial / ${#shot_values[@]}))
adapt_steps=${adapt_steps_values[$((trial % ${#adapt_steps_values[@]}))]}
trial=$((trial / ${#adapt_steps_values[@]}))
out_dim=${out_dim_values[$((trial % ${#out_dim_values[@]}))]}
trial=$((trial / ${#out_dim_values[@]}))
granularity=${granularity_values[$((trial % ${#granularity_values[@]}))]}

trial=$((trial / ${#granularity_values[@]}))
fo=${fo_values[$((trial % ${#fo_values[@]}))]}

python pMAML.py --file ${file} --agg ${type} --alg ${alg} --epochs 100 --train --adapt_lr ${adapt_lr} --lr ${lr} --shots ${shots} --adapt_steps ${adapt_steps} --out_dim ${out_dim} --granularity ${granularity} --${fo}