LR=0.0005
EPOCHS=20
BATCH_SIZE=128
OUTPUT_BASE="siamese_fbanks_saved/"
TRAIN_DATA="fbannks_train"
TEST_DATA="fbannks_test"

for NUM_LAYERS in 2 3 4 5 6
do
  PRETRAINED_MODEL_PATH="saved_models_cross_entropy/${NUM_LAYERS}/"
  OUTPUT_MODEL_PATH="${OUTPUT_BASE}${NUM_LAYERS}/"
  
  echo "Running training with num_layers=${NUM_LAYERS}, pretrained_model_path=${PRETRAINED_MODEL_PATH}, output_model_path=${OUTPUT_MODEL_PATH}"

  python3 stage2_finetune.py \
    --num_layers ${NUM_LAYERS} \
    --lr ${LR} \
    --epochs ${EPOCHS} \
    --batch_size ${BATCH_SIZE} \
    --pretrained_model_path ${PRETRAINED_MODEL_PATH} \
    --output_model_path ${OUTPUT_MODEL_PATH} \
    --train_data ${TRAIN_DATA} \
    --test_data ${TEST_DATA}
  
  echo "Finished training with num_layers=${NUM_LAYERS}"
done