I think we are missing entries in the documentation from these additions. The ViT models small/large with/without LoRA are not listed here: https://mednet.readthedocs.io/en/latest/models.html The APTOS3 dataset is not listed here: https://mednet.readthedocs.io/en/latest/databases/index.html @rpulvirenti: could you please add these?