Training in progress, epoch 1

config.json

@@ -1,49 +1,33 @@
 {
-  "_name_or_path": "priyankloco/resnet-101-finetuned_resnet101-cnn-autotags",
+  "_name_or_path": "akahana/vit-base-cats-vs-dogs",
   "architectures": [
-    "ResNetForImageClassification"
-  ],
-  "depths": [
-    3,
-    4,
-    23,
-    3
-  ],
-  "downsample_in_bottleneck": false,
-  "downsample_in_first_stage": false,
-  "embedding_size": 64,
-  "hidden_act": "relu",
-  "hidden_sizes": [
-    256,
-    512,
-    1024,
-    2048
+    "ViTForImageClassification"
   ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "finetuning_task": "image-classification",
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
   "id2label": {
     "0": "cat",
     "1": "dog"
   },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
   "label2id": {
     "cat": "0",
     "dog": "1"
   },
-  "layer_type": "bottleneck",
-  "model_type": "resnet",
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
   "num_channels": 3,
-  "out_features": [
-    "stage4"
-  ],
-  "out_indices": [
-    4
-  ],
+  "num_hidden_layers": 12,
+  "patch_size": 16,
   "problem_type": "single_label_classification",
-  "stage_names": [
-    "stem",
-    "stage1",
-    "stage2",
-    "stage3",
-    "stage4"
-  ],
+  "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.37.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:29d493cd293919c301bbd06a597fe4abb8a66c2e100c80de8f93017c841a12cd
-size 170521528
+oid sha256:62b9c3cc5c41d5710952fbe498e0e7936baa46af95ac21905185db44da39bae0
+size 343223968

preprocessor_config.json

@@ -1,22 +1,22 @@
 {
-  "crop_pct": 0.875,
   "do_normalize": true,
   "do_rescale": true,
   "do_resize": true,
   "image_mean": [
-    0.485,
-    0.456,
-    0.406
+    0.5,
+    0.5,
+    0.5
   ],
-  "image_processor_type": "ConvNextFeatureExtractor",
+  "image_processor_type": "ViTFeatureExtractor",
   "image_std": [
-    0.229,
-    0.224,
-    0.225
+    0.5,
+    0.5,
+    0.5
   ],
-  "resample": 3,
+  "resample": 2,
   "rescale_factor": 0.00392156862745098,
   "size": {
-    "shortest_edge": 224
+    "height": 224,
+    "width": 224
   }
 }

runs/Feb24_07-18-46_21999e1c2e14/events.out.tfevents.1708759126.21999e1c2e14.1704.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:09ce4939813c296d2e1b04516e4cd744d3115801dff11afcafaa14550767943b
-size 5440
+oid sha256:add3d2eeb6b558d5a8696f548a9063911ae999312e394495ada23a8724340a3a
+size 6222

runs/Feb24_07-33-28_21999e1c2e14/events.out.tfevents.1708760009.21999e1c2e14.1704.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e5c4eac54df04c46e91bb3f7ad074863f36896b8e04572388fb3cb1928d811a5
+size 4468

runs/Feb24_07-33-54_21999e1c2e14/events.out.tfevents.1708760034.21999e1c2e14.1704.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:30481220b1dba012e534c1b92237905c494b5c3e92352ce06b600410b5c881b1
+size 4468

runs/Feb24_07-35-38_21999e1c2e14/events.out.tfevents.1708760139.21999e1c2e14.1704.3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ac56d90bd10c1b5f7263e763bd9b65d9f15bdd3a99a6ee39c998cb7352a62e6
+size 4468

runs/Feb24_07-37-46_21999e1c2e14/events.out.tfevents.1708760267.21999e1c2e14.1704.4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b2fd5de513955af82e76f9f8fb1c947a179ed94c11ecdde496fb3e4bfbcec22
+size 4896

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:48010a137597ad6d800f89718cb4525aa0b6d1e0bda6592833985941e70cb295
+oid sha256:57a940d8e6ec76bb212d9ca799e9fe486f20d7d4a3ca6702e6de75add9f82fc3
 size 4728