Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -28,6 +28,12 @@ Released as supporting material for an accompanying paper.
|
|
| 28 |
used only for an auxiliary contrastive image-text alignment loss during training.
|
| 29 |
It is **not used at inference time** — `forward()` only calls `image_encoder` and `cls_head`.
|
| 30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
## Files in this repo
|
| 32 |
|
| 33 |
| File | Purpose |
|
|
@@ -110,8 +116,9 @@ pip install -r requirements-train.txt
|
|
| 110 |
python train.py --tfrecord-path /path/to/.tfrecord --output-dir ./runs/my_run
|
| 111 |
```
|
| 112 |
|
| 113 |
-
**Data access:**
|
| 114 |
-
|
|
|
|
| 115 |
|
| 116 |
**Note on training-set size:** training draws 1000 samples per epoch, with replacement and augmentation, from only 112 unique images — not 1000 unique examples.
|
| 117 |
|
|
|
|
| 28 |
used only for an auxiliary contrastive image-text alignment loss during training.
|
| 29 |
It is **not used at inference time** — `forward()` only calls `image_encoder` and `cls_head`.
|
| 30 |
|
| 31 |
+
## Dataset
|
| 32 |
+
|
| 33 |
+
Dataset: [kavin-aravindhan/oct-wamd](https://huggingface.co/datasets/kavin-aravindhan/oct-wamd).
|
| 34 |
+
Public access is pending IRB approval — for interim access, contact Kavin
|
| 35 |
+
Aravindhan (kr3131@columbia.edu).
|
| 36 |
+
|
| 37 |
## Files in this repo
|
| 38 |
|
| 39 |
| File | Purpose |
|
|
|
|
| 116 |
python train.py --tfrecord-path /path/to/.tfrecord --output-dir ./runs/my_run
|
| 117 |
```
|
| 118 |
|
| 119 |
+
**Data access:** see [Dataset](#dataset) above — the training set (112
|
| 120 |
+
labeled images with clinical-finding captions) is hosted as a private HF
|
| 121 |
+
dataset pending IRB approval.
|
| 122 |
|
| 123 |
**Note on training-set size:** training draws 1000 samples per epoch, with replacement and augmentation, from only 112 unique images — not 1000 unique examples.
|
| 124 |
|