mednet.config.classify.models

Modules

alexnet	AlexNet network architecture, to be trained from scratch.
alexnet_pretrained	AlexNet network architecture, pre-trained with ImageNet weights.
cnn3d	Simple CNN for 3D image classification, to be trained from scratch.
densenet	DenseNet, to be trained from scratch.
densenet_pretrained	DenseNet, to be fine-tuned.
logistic_regression	Logistic regression classifier, to be trained from scratch.
mlp	Feedforward shallow MLP for binary classification.
pasa	Simple CNN for Tuberculosis Detection, to be trained from scratch.
vit_large	ViT-Large, to be fine-tuned.
vit_large_lora	ViT-Large model configuration for fine-tuning with LoRA.
vit_small	ViT-Small, to be fine-tuned.
vit_small_lora	ViT-Small model configuration for fine-tuning with LoRA.