airkjw/RETFound
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Incorporate DINOv3, DINOv2

Browse Source
This commit is contained in:
rmaphoh
2025-08-31 18:07:38 +01:00
parent 409f7b6167
commit 7489af0620
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.sh
+7 -6
View File
@@ -1,17 +1,18 @@
# ==== Model settings ==== # ==== Model settings ====
# adaptation {finetune,lp}
ADAPTATION="finetune" ADAPTATION="finetune"
MODEL="Dinov2" MODEL="RETFound_dinov2"
MODEL_ARCH="dinov2_vitl14" MODEL_ARCH="retfound_dinov2"
FINETUNE="dinov2_vitl14_pretrain.pth" FINETUNE="RETFound_dinov2_meh"
# ==== Data settings ==== # ==== Data settings ====
# change the dataset name and corresponding class number
DATASET="MESSIDOR2" DATASET="MESSIDOR2"
NUM_CLASS=5 NUM_CLASS=5
data_path="./${DATASET}"
data_path="/home/jupyter/public_dataset/${DATASET}"
task="${MODEL_ARCH}_${DATASET}_${ADAPTATION}" 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 "${MODEL}" \
--model_arch "${MODEL_ARCH}" \ --model_arch "${MODEL_ARCH}" \
--finetune "${FINETUNE}" \ --finetune "${FINETUNE}" \