taokz / BiomedGPT

BiomedGPT: A Generalist Vision-Language Foundation Model for Diverse Biomedical Tasks
https://www.nature.com/articles/s41591-024-03185-2
Apache License 2.0
515 stars 59 forks source link

where to find checkpoint_best.pt mentioned in evaluate_medmnist.sh #54

Open Trista1105 opened 13 hours ago

Trista1105 commented 13 hours ago

Hello, in the file evaluate_medmnist.sh, path=../../checkpoints/tuned_checkpoints/MedMNIST/base/${dataset}/25000_1000_5e-5_256/checkpoint_best.pt

Where to find this file for MedMNIST datasets?

taokz commented 10 hours ago

Hi, I don’t have enough cloud storage to share the checkpoint directly. However, you can fine-tune the model using the provided pre-trained checkpoint. You’ll find the necessary script in scripts/image_cls (just need to modify the gpu setup, e.g., CUDA_VISIBLE_DEVICES) and the checkpoints listed in checkpoints.md. Let me know if you have any questions!