rmaphoh
28 lines
725 B
Bash
28 lines
725 B
Bash
# ==== Model settings ====
|
|
# adaptation {finetune,lp}
|
|
ADAPTATION="finetune"
|
|
MODEL="RETFound_dinov2"
|
|
MODEL_ARCH="retfound_dinov2"
|
|
FINETUNE="RETFound_dinov2_meh"
|
|
|
|
# ==== Data settings ====
|
|
# change the dataset name and corresponding class number
|
|
DATASET="MESSIDOR2"
|
|
NUM_CLASS=5
|
|
data_path="./${DATASET}"
|
|
task="${MODEL_ARCH}_${DATASET}_${ADAPTATION}"
|
|
|
|
torchrun --nproc_per_node=1 --master_port=48766 main_finetune.py \
|
|
--model "${MODEL}" \
|
|
--model_arch "${MODEL_ARCH}" \
|
|
--finetune "${FINETUNE}" \
|
|
--savemodel \
|
|
--global_pool \
|
|
--batch_size 24 \
|
|
--world_size 1 \
|
|
--epochs 50 \
|
|
--nb_classes "${NUM_CLASS}" \
|
|
--data_path "${data_path}" \
|
|
--input_size 224 \
|
|
--task "${task}" \
|
|
--adaptation "${ADAPTATION}" |