Skip to content

Commit 6eb7d83

Browse files
authored
Update classifier.py
1 parent db54cdf commit 6eb7d83

1 file changed

Lines changed: 0 additions & 10 deletions

File tree

models/classifier.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,3 @@
1-
"""
2-
Multi-label classification head with MC Dropout.
3-
4-
Architecture:
5-
BioMedCLIP ViT-B/16 → LayerNorm → Dropout → Linear(512, 256)
6-
→ GELU → Dropout → Linear(256, 14)
7-
8-
The double-dropout design enables Monte Carlo uncertainty estimation at
9-
inference time by keeping dropout active (model.train() mode).
10-
"""
111

122
from __future__ import annotations
133

0 commit comments

Comments
 (0)