File size: 1,321 Bytes
8a75cee
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
    "_name_or_path": "microsoft/resnet-18",
    "architectures": [
      "CifNetForImageClassification"
    ],
    "embedding_kwargs": {
      "__comment": "First hidden layer size",
      "embedding_size": 8,
      "embedding_kernel_size_1": 7,
      "embedding_kernel_size_2": 2,
      "embedding_stride_1": 2,
      "embedding_stride_2": 2
    },
    "attention_kwargs": {
      "attn_channels": 8,
      "num_heads": 4,
      "attn_kernel_size": 3,
      "attn_stride": 1,
      "attention_bias": true,
      "attention_dropout": 0.1,
      "max_position_embeddings": null
    },
    "depths": [
      8,
      12,
      28
    ],
    "activation": "silu",
    "hidden_sizes": [
      16,
      32,
      64
    ],
    "id2label": {
      "0": "airplane",
      "1": "automobile",
      "2": "bird",
      "3": "cat",
      "4": "deer",
      "5": "dog",
      "6": "frog",
      "7": "horse",
      "8": "ship",
      "9": "truck"
    },
    "label2id": {
      "airplane": "0",
      "automobile": "1",
      "bird": "2",
      "cat": "3",
      "deer": "4",
      "dog": "5",
      "frog": "6",
      "horse": "7",
      "ship": "8",
      "truck": "9"
    },
    "layer_type": "attention",
    "model_type": "resnet",
    "num_channels": 3,
    "torch_dtype": "float32",
    "transformers_version": "4.38.2"
  }