Model save

config.json

@@ -1,34 +1,36 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "microsoft/resnet-50",
   "architectures": [
-    "ViTForImageClassification"
+    "ResNetForImageClassification"
   ],
-  "attention_probs_dropout_prob": 0.0,
-  "encoder_stride": 16,
+  "depths": [
+    3,
+    4,
+    6,
+    3
+  ],
+  "downsample_in_first_stage": false,
+  "embedding_size": 64,
   "finetuning_task": "image-classification",
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.0,
-  "hidden_size": 768,
+  "hidden_act": "relu",
+  "hidden_sizes": [
+    256,
+    512,
+    1024,
+    2048
+  ],
   "id2label": {
-    "0": "angular_leaf_spot",
-    "1": "bean_rust",
-    "2": "healthy"
+    "0": "cats",
+    "1": "dogs"
   },
-  "image_size": 224,
-  "initializer_range": 0.02,
-  "intermediate_size": 3072,
   "label2id": {
-    "angular_leaf_spot": "0",
-    "bean_rust": "1",
-    "healthy": "2"
+    "cats": "0",
+    "dogs": "1"
   },
-  "layer_norm_eps": 1e-12,
-  "model_type": "vit",
-  "num_attention_heads": 12,
+  "layer_type": "bottleneck",
+  "model_type": "resnet",
   "num_channels": 3,
-  "num_hidden_layers": 12,
-  "patch_size": 16,
-  "qkv_bias": true,
+  "problem_type": "single_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.22.0.dev0"
 }

preprocessor_config.json

@@ -1,17 +1,18 @@
 {
+  "crop_pct": 0.875,
   "do_normalize": true,
   "do_resize": true,
-  "feature_extractor_type": "ViTFeatureExtractor",
+  "feature_extractor_type": "ConvNextFeatureExtractor",
   "image_mean": [
-    0.5,
-    0.5,
-    0.5
+    0.485,
+    0.456,
+    0.406
   ],
   "image_std": [
-    0.5,
-    0.5,
-    0.5
+    0.229,
+    0.224,
+    0.225
   ],
-  "resample": 2,
+  "resample": 3,
   "size": 224
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:505f5d3eed83801cc8b6ec6957a0959c5354ee83105b59fb6111b7d6f3d26104
-size 343270065
+oid sha256:632f98ee60fb0af9af8a0474bf6338186a1cb212ef578627d0816a494e58f594
+size 94372673

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5d9db89873c68fe1c98704acd87155c616e259f10e1d3995be58c90b4fd2c4c5
+oid sha256:57f9b955c2006f19e884f2ba2127bae2b54b3378da247d1c0800c9a874345d69
 size 3375