Add weights and configs

README.md

@@ -49,8 +49,8 @@ import requests
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 
-feature_extractor = AutoImageProcessor.from_pretrained("shi-labs/dinat-tiny-in1k-224")
-model = DinatForImageClassification.from_pretrained("shi-labs/dinat-tiny-in1k-224")
+feature_extractor = AutoImageProcessor.from_pretrained("shi-labs/dinat-mini-in1k-224")
+model = DinatForImageClassification.from_pretrained("shi-labs/dinat-mini-in1k-224")
 
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)

config.json

@@ -9,10 +9,39 @@
     6,
     5
   ],
+  "dilations": [
+    [
+      1,
+      8,
+      1
+    ],
+    [
+      1,
+      4,
+      1,
+      4
+    ],
+    [
+      1,
+      2,
+      1,
+      2,
+      1,
+      2
+    ],
+    [
+      1,
+      1,
+      1,
+      1,
+      1
+    ]
+  ],
   "drop_path_rate": 0.2,
   "embed_dim": 64,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
+  "hidden_size": 512,
   "id2label": {
     "0": "tench, Tinca tinca",
     "1": "goldfish, Carassius auratus",
@@ -1015,8 +1044,8 @@
     "998": "ear, spike, capitulum",
     "999": "toilet tissue, toilet paper, bathroom tissue"
   },
-  "image_size": 224,
   "initializer_range": 0.02,
+  "kernel_size": 7,
   "label2id": {
     "Afghan hound, Afghan": 160,
     "African chameleon, Chamaeleo chamaeleon": 47,
@@ -2019,6 +2048,7 @@
     "zucchini, courgette": 939
   },
   "layer_norm_eps": 1e-05,
+  "layer_scale_init_value": 0.0,
   "mlp_ratio": 3.0,
   "model_type": "dinat",
   "num_channels": 3,
@@ -2028,11 +2058,10 @@
     8,
     16
   ],
+  "num_layers": 4,
   "patch_size": 4,
   "path_norm": true,
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.25.0.dev0",
-  "kernel_size": 7,
-  "dilations": [[1, 8, 1], [1, 4, 1, 4], [1, 2, 1, 2, 1, 2], [1, 1, 1, 1, 1]]
+  "transformers_version": "4.25.0.dev0"
 }