#!/bin/bash
# run_spherec_inat17.sh

DIR=../models/sphere2vec_sphereC/

ENC=Sphere2Vec-sphereC

DATA=inat_2017
# DATA=inat_2018
# DATA=nabirds
# DATA=birdsnap
# META=orig_meta
META=ebird_meta
EVALDATA=val

DEVICE=cuda:1

LR=0.0005
LAYER=1
HIDDIM=1024
FREQ=32
MINR=0.001
MAXR=1
EPOCH=100


ACT=relu
RATIO=1.0


for x in inat_2017,ebird_meta,val #birdsnap,orig_meta,test  birdsnap,ebird_meta,test  nabirds,ebird_meta,test   #inat_2018,ebird_meta,val
do
    IFS=',' read DATA  META  EVALDATA <<< "${x}"
    for LR in 0.0001 #0.0005 0.001 0.002 0.005
    do
        for MINR in 0.01 #0.001 0.0005 0.0001
        do  
            for ACT in relu leakyrelu
            do
                python3 main.py \
                    --spa_enc_type $ENC \
                    --meta_type $META\
                    --dataset $DATA \
                    --eval_split $EVALDATA \
                    --frequency_num $FREQ \
                    --max_radius $MAXR \
                    --min_radius $MINR \
                    --num_hidden_layer $LAYER \
                    --hidden_dim $HIDDIM \
                    --spa_f_act $ACT \
                    --unsuper_lr 0.1 \
                    --lr $LR \
                    --model_dir $DIR \
                    --num_epochs $EPOCH \
                    --train_sample_ratio $RATIO \
                    --device $DEVICE
            done
        done
    done
done
