Open Trista1105 opened 13 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!
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?