File size: 1,985 Bytes
5043776 490c430 5043776 d16d317 5043776 d16d317 490c430 5043776 490c430 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
---
library_name: transformers
tags:
- generated_from_trainer
model-index:
- name: SkLIP-masked
results: []
license: mit
datasets:
- joshuachou/SkinCAP
language:
- en
base_model:
- allenai/scibert_scivocab_uncased
- openai/clip-vit-base-patch32
pipeline_tag: feature-extraction
---
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->
# SkLIP
SkLIP (Skin CLIP) is a hybrid CLIP model finetuned on the [SkinCAP](https://hf.rst.im/datasets/joshuachou/SkinCAP), a multi-modal dermatology dataset annotated
with rich medical captions. It is built witha SciBERT text encoder and the pre-trained CLIP-32 vision encoder.
## Model description
More information needed
## Intended uses & limitations
More information needed
## Training and evaluation data
More information needed
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 1e-05
- train_batch_size: 64
- eval_batch_size: 64
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: cosine
- lr_scheduler_warmup_ratio: 0.01
- num_epochs: 10
### Training results
| Training Loss | Epoch | Step | Validation Loss |
|:-------------:|:-----:|:----:|:---------------:|
| 4.2018 | 1.0 | 57 | 4.1344 |
| 4.1697 | 2.0 | 114 | 4.1298 |
| 4.1668 | 3.0 | 171 | 4.1276 |
| 4.164 | 4.0 | 228 | 4.1263 |
| 4.158 | 5.0 | 285 | 4.1253 |
| 4.1583 | 6.0 | 342 | 4.1246 |
| 4.1569 | 7.0 | 399 | 4.1243 |
| 4.1575 | 8.0 | 456 | 4.1241 |
| 4.1564 | 9.0 | 513 | 4.1240 |
| 4.1604 | 10.0 | 570 | 4.1240 |
### Framework versions
- Transformers 4.45.2
- Pytorch 2.1.0+cu118
- Datasets 3.0.1
- Tokenizers 0.20.1 |