Hugging Face's logo

ufal
/
vit-historical-page

Image Classification
Transformers
Safetensors
timm
timm_wrapper
page
classification
Model card Files
xet
Community
k4tel commited on
Commit
bf0b5db
·
verified ·
1 Parent(s): b141f06

Upload TimmWrapperForImageClassification

Browse files
Files changed (3) hide show
  1. README.md +1 -0
  2. config.json +52 -43
  3. model.safetensors +2 -2
README.md CHANGED
@@ -3,6 +3,7 @@ library_name: transformers
3
  tags:
4
  - page
5
  - classification
 
6
  base_model:
7
  - google/vit-base-patch16-224
8
  - google/vit-base-patch16-384
 
3
  tags:
4
  - page
5
  - classification
6
+ - timm
7
  base_model:
8
  - google/vit-base-patch16-224
9
  - google/vit-base-patch16-384
config.json CHANGED
@@ -1,50 +1,59 @@
1
  {
 
2
  "architectures": [
3
- "ViTForImageClassification"
4
  ],
5
- "attention_probs_dropout_prob": 0.0,
6
- "encoder_stride": 16,
7
- "hidden_act": "gelu",
8
- "hidden_dropout_prob": 0.0,
9
- "hidden_size": 1024,
10
- "id2label": {
11
- "0": "LABEL_0",
12
- "1": "LABEL_1",
13
- "2": "LABEL_2",
14
- "3": "LABEL_3",
15
- "4": "LABEL_4",
16
- "5": "LABEL_5",
17
- "6": "LABEL_6",
18
- "7": "LABEL_7",
19
- "8": "LABEL_8",
20
- "9": "LABEL_9",
21
- "10": "LABEL_10"
22
- },
23
- "image_size": 384,
24
  "initializer_range": 0.02,
25
- "intermediate_size": 4096,
26
- "label2id": {
27
- "LABEL_0": 0,
28
- "LABEL_1": 1,
29
- "LABEL_10": 10,
30
- "LABEL_2": 2,
31
- "LABEL_3": 3,
32
- "LABEL_4": 4,
33
- "LABEL_5": 5,
34
- "LABEL_6": 6,
35
- "LABEL_7": 7,
36
- "LABEL_8": 8,
37
- "LABEL_9": 9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
38
  },
39
- "layer_norm_eps": 1e-12,
40
- "model_type": "vit",
41
- "num_attention_heads": 16,
42
- "num_channels": 3,
43
- "num_hidden_layers": 24,
44
- "patch_size": 16,
45
- "pooler_act": "tanh",
46
- "pooler_output_size": 1024,
47
- "qkv_bias": true,
48
  "torch_dtype": "float32",
49
- "transformers_version": "4.52.4"
50
  }
 
1
  {
2
+ "architecture": "regnety_160",
3
  "architectures": [
4
+ "TimmWrapperForImageClassification"
5
  ],
6
+ "do_pooling": true,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  "initializer_range": 0.02,
8
+ "label_names": [
9
+ "LABEL_0",
10
+ "LABEL_1",
11
+ "LABEL_2",
12
+ "LABEL_3",
13
+ "LABEL_4",
14
+ "LABEL_5",
15
+ "LABEL_6",
16
+ "LABEL_7",
17
+ "LABEL_8",
18
+ "LABEL_9",
19
+ "LABEL_10"
20
+ ],
21
+ "model_args": null,
22
+ "model_type": "timm_wrapper",
23
+ "num_classes": 11,
24
+ "num_features": 3024,
25
+ "pretrained_cfg": {
26
+ "classifier": "head.fc",
27
+ "crop_mode": "center",
28
+ "crop_pct": 1.0,
29
+ "custom_load": false,
30
+ "first_conv": "stem.conv",
31
+ "fixed_input_size": false,
32
+ "input_size": [
33
+ 3,
34
+ 384,
35
+ 384
36
+ ],
37
+ "interpolation": "bicubic",
38
+ "license": "cc-by-nc-4.0",
39
+ "mean": [
40
+ 0.485,
41
+ 0.456,
42
+ 0.406
43
+ ],
44
+ "origin_url": "https://github.com/pytorch/vision",
45
+ "pool_size": [
46
+ 12,
47
+ 12
48
+ ],
49
+ "std": [
50
+ 0.229,
51
+ 0.224,
52
+ 0.225
53
+ ],
54
+ "tag": "swag_ft_in1k"
55
  },
56
+ "problem_type": "multi_label_classification",
 
 
 
 
 
 
 
 
57
  "torch_dtype": "float32",
58
+ "transformers_version": "4.53.1"
59
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7d6484924e55134512e863689636f8e8936d92e57f612db69858e72f82a63f25
3
- size 1214854652
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b18e89783dfd8e36ae75da62fc67a657d065f66bed9cc9d54e9771993a50ea1
3
+ size 322925148