diff --git a/train.sh b/train.sh
index 0a3ee46..7302d5a 100644
--- a/train.sh
+++ b/train.sh
@@ -1,17 +1,18 @@
 # ==== Model settings ====
+# adaptation {finetune,lp}
 ADAPTATION="finetune"
-MODEL="Dinov2"
-MODEL_ARCH="dinov2_vitl14"
-FINETUNE="dinov2_vitl14_pretrain.pth"
+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="/home/jupyter/public_dataset/${DATASET}"
+data_path="./${DATASET}"
 task="${MODEL_ARCH}_${DATASET}_${ADAPTATION}"
 
-CUDA_VISIBLE_DEVICES=1 torchrun --nproc_per_node=1 --master_port=48766 main_finetune.py \
+torchrun --nproc_per_node=1 --master_port=48766 main_finetune.py \
   --model "${MODEL}" \
   --model_arch "${MODEL_ARCH}" \
   --finetune "${FINETUNE}" \
@@ -24,4 +25,4 @@ CUDA_VISIBLE_DEVICES=1 torchrun --nproc_per_node=1 --master_port=48766 main_fine
   --data_path "${data_path}" \
   --input_size 224 \
   --task "${task}" \
-  --adaptation "${ADAPTATION}" 
+  --adaptation "${ADAPTATION}" 
\ No newline at end of file