Anwarkh1 commited on
Commit
632bf0f
1 Parent(s): 8926e6f

Upload ViTForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +0 -18
  2. model.safetensors +1 -1
config.json CHANGED
@@ -8,27 +8,9 @@
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
- "id2label": {
12
- "0": "benign_keratosis-like_lesions",
13
- "1": "basal_cell_carcinoma",
14
- "2": "actinic_keratoses",
15
- "3": "vascular_lesions",
16
- "4": "melanocytic_Nevi",
17
- "5": "melanoma",
18
- "6": "dermatofibroma"
19
- },
20
  "image_size": 224,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
- "label2id": {
24
- "benign_keratosis-like_lesions": 0,
25
- "basal_cell_carcinoma": 1,
26
- "actinic_keratoses": 2,
27
- "vascular_lesions": 3,
28
- "melanocytic_Nevi": 4,
29
- "melanoma": 5,
30
- "dermatofibroma": 6
31
- },
32
  "layer_norm_eps": 1e-12,
33
  "model_type": "vit",
34
  "num_attention_heads": 12,
 
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
 
 
 
 
 
 
 
 
 
11
  "image_size": 224,
12
  "initializer_range": 0.02,
13
  "intermediate_size": 3072,
 
 
 
 
 
 
 
 
 
14
  "layer_norm_eps": 1e-12,
15
  "model_type": "vit",
16
  "num_attention_heads": 12,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:501d946d1111fd3577716788f21f0b02b228d2e0c4eae301cdfad47a9246bdff
3
  size 343239356
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b8136bbbdd085cc97f69bdef01bd0c4c54262cf3ba44f5a94ebc9c8c54e85d60
3
  size 343239356