From 70b56d34ec06ec3b77cb49c86f6ee9c65245ccdf Mon Sep 17 00:00:00 2001 From: rmaphoh Date: Wed, 19 Feb 2025 13:03:38 +0000 Subject: [PATCH] major package upgrade&new weights --- README.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index e854b84..543ddf8 100644 --- a/README.md +++ b/README.md @@ -120,7 +120,10 @@ huggingface-cli login --token YOUR_HUGGINGFACE_TOKEN model can be "RETFound_mae" or "RETFound_dinov2" ``` ``` -finetune can be "RETFound_mae_natureOCT", "RETFound_mae_natureCFP", "RETFound_mae_meh", "RETFound_mae_shanghai", "RETFound_dinov2_meh", and "RETFound_dinov2_shanghai". +when model is "RETFound_mae", finetune can be "RETFound_mae_natureOCT", "RETFound_mae_natureCFP", "RETFound_mae_meh", "RETFound_mae_shanghai". +``` +``` +when model is "RETFound_dinov2", finetune can be "RETFound_dinov2_meh", and "RETFound_dinov2_shanghai". ``` ``` torchrun --nproc_per_node=1 --master_port=48798 main_finetune.py \