# ROOD, switching from VSA to no VSA
export CUDA_VISIBLE_DEVICES="0"

# SET THE CORRECT MODEL_CHECKPOINTS TO MATCH EACH CONFIGURATION

# No HRR
python3 BERTHA_Finetune_Disease_Prediction.py --snomed_group=ignore --dp_type=atomic --dp_composition=snomed_all \
    --num_trials=1 --epochs=10 \
    --model_checkpoint="[SET HERE]"

# HRR
python3 BERTHA_Finetune_Disease_Prediction.py --use_vsa --snomed_group=ignore --dp_type=atomic --dp_composition=snomed_all \
    --num_trials=10 --epochs=10 \
    --model_checkpoint="[SET HERE]"

# HRR Add
python3 BERTHA_Finetune_Disease_Prediction.py --use_vsa --snomed_group=ignore --dp_type=atomic --dp_composition=snomed_all \
    --num_trials=10 --epochs=10 \
    --model_checkpoint="[SET HERE]"

# HRR CAT
python3 BERTHA_Finetune_Disease_Prediction.py --use_vsa --snomed_group=ignore --dp_type=atomic --dp_composition=snomed_all \
    --num_trials=10 --epochs=10 \
    --model_checkpoint="[SET HERE]"