Training in progress epoch 0

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: google/vit-large-patch16-224-in21k
+base_model: google/vit-base-patch16-384
 tags:
 - generated_from_keras_callback
 model-index:
@@ -13,12 +13,12 @@ probably proofread and complete it, then remove this comment. -->
 
 # Prahas10/roof-large
 
-This model is a fine-tuned version of [google/vit-large-patch16-224-in21k](https://huggingface.co/google/vit-large-patch16-224-in21k) on an unknown dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-384](https://huggingface.co/google/vit-base-patch16-384) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.1559
-- Validation Loss: 0.2924
-- Train Accuracy: 0.8909
-- Epoch: 8
+- Train Loss: 1.6423
+- Validation Loss: 1.0616
+- Train Accuracy: 0.7091
+- Epoch: 0
 
 ## Model description
 
@@ -44,15 +44,7 @@ The following hyperparameters were used during training:
 
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 1.8354     | 1.1693          | 0.6364         | 0     |
-| 0.6363     | 0.5030          | 0.8455         | 1     |
-| 0.4077     | 0.3187          | 0.9091         | 2     |
-| 0.2877     | 0.4201          | 0.8273         | 3     |
-| 0.3060     | 0.6308          | 0.8364         | 4     |
-| 0.3030     | 0.2840          | 0.9091         | 5     |
-| 0.1304     | 0.1397          | 0.9545         | 6     |
-| 0.3800     | 0.4405          | 0.8455         | 7     |
-| 0.1559     | 0.2924          | 0.8909         | 8     |
+| 1.6423     | 1.0616          | 0.7091         | 0     |
 
 
 ### Framework versions

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-large-patch16-224-in21k",
+  "_name_or_path": "google/vit-base-patch16-384",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -7,7 +7,7 @@
   "encoder_stride": 16,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
-  "hidden_size": 1024,
+  "hidden_size": 768,
   "id2label": {
     "0": "cteed_landmark_driftwood_shade",
     "1": "cteed_landmark_driftwood_sun",
@@ -25,9 +25,9 @@
     "8": "iko_dynasty_cornerstone_sun",
     "9": "iko_dynasty_granite_black_shade"
   },
-  "image_size": 224,
+  "image_size": 384,
   "initializer_range": 0.02,
-  "intermediate_size": 4096,
+  "intermediate_size": 3072,
   "label2id": {
     "cteed_landmark_driftwood_shade": "0",
     "cteed_landmark_driftwood_sun": "1",
@@ -47,9 +47,9 @@
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
-  "num_attention_heads": 16,
+  "num_attention_heads": 12,
   "num_channels": 3,
-  "num_hidden_layers": 24,
+  "num_hidden_layers": 12,
   "patch_size": 16,
   "qkv_bias": true,
   "transformers_version": "4.38.2"

preprocessor_config.json

@@ -16,7 +16,7 @@
   "resample": 2,
   "rescale_factor": 0.00392156862745098,
   "size": {
-    "height": 224,
-    "width": 224
+    "height": 384,
+    "width": 384
   }
 }

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:809e00f453d3c66b4856e10a2c954417b439883d3519667b95cfb38bd7e28c92
-size 1213816912
+oid sha256:0f6e05741eb47ffdbac2a905a17d7a64b3e946fe3e5caff0a470827ffcce0e42
+size 344677048