#!/bin/bash

export WANDB_API_KEY="0054fda6ed25eace6e3c37c9042258f123cccf4c"

source /home/XXXX-11/miniforge3/bin/activate lmfact
module load cuda/12.1
cd /is/cluster/XXXX-11/forecasting-rl/qgen/

# Run the classifier with accelerate
accelerate launch classifier.py \
    --data_path "$1" \
    --model_name "$2" \
    --output_dir "$3" \
    --wandb_project "$4" \
    --wandb_run_name "$5" \
    --train_ratio "$6" \
    --test_ratio "$7" \
    --num_train_epochs "$8" \
    --gradacc_steps "$9" \
    ${10} \
    ${11}

conda deactivate