search

duyhominhnguyen / LVM-Med

[NeurIPS 2023] Release LMV-Med pre-trained models
188 stars 24 forks source link

Classification tasks with ViT #12

Closed sahil02235 closed 9 months ago

sahil02235 commented 10 months ago

Hi @duyhominhnguyen Thanks for the great work. Can you please provide the fine-tuning code for classification tasks with ViT model.

Thanks and regards, Sahil

duyhominhnguyen commented 10 months ago

Hi @sahil02235 do you mean code to fine-tuning with VIT on classification with our models?

sahil02235 commented 10 months ago

Yes, exactly. I have also requested for pre-training code. Can you please check.

nghiemkythu commented 10 months ago

Hi @sahil02235, I have uploaded the version of classification using VIT and files .yml to run VIT for FGADR dataset. The code use VIT for classification is in the folder "classification_VIT".

If you want to run this version, you can run the same as the R50 version except that you change the key word "R50" to "VIT"