Question regarding model checkpoints

mazurowski-lab / finetune-SAM

This is an official repo for fine-tuning SAM to customized medical images.

https://arxiv.org/abs/2404.09957

Apache License 2.0

136 stars 22 forks source link

Question regarding model checkpoints #11

Closed anwai98 closed 5 months ago

anwai98 commented 5 months ago

Hi team,

Thanks for the really cool work, and also for open-sourcing your scripts.

I have a question: have you released the model checkpoints from your experiments (related to the different tasks performed for finetuning / self-supervised pretraining)? Maybe I totally overlooked it, I couldn't spot the download links. Would be happy to get access to them, if possible.

Thanks in advance!

haoyudong-97 commented 5 months ago

We have the checkpoints for self-supervised pretraining at Step 3C (https://drive.google.com/drive/folders/1JAoy-Mh5QgxXsjWtQhMjOX16dN1kytLQ); we did not provide the checkpoints for self-supervised pertaining + fine-tuning since we have too many fine-tuning tasks.