model.sh

#!/bin/bash

input_csv_path="$1"
output_csv_path="$2"
max_depth=6
device='cuda:0'
model_path_list=( 
    "../../save_model_ckp/gda_infoNCE_2024_GPU3090" \
)

cd ../src/finetune/
for save_model_path in ${model_path_list[@]}; do
        num_leaves=$((2**($max_depth-1)))
        python finetune.py \
                --input_csv_path $input_csv_path \
                --output_csv_path $output_csv_path \
                --save_model_path $save_model_path \
                --device $device \
                --batch_size 128 \
                --step "300" \
                --use_pooled \
                --num_leaves $num_leaves \
                --max_depth $max_depth
done