End of training

README.md

@@ -15,9 +15,9 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 2.3043
-- Validation Loss: 2.3868
-- Train Accuracy: 0.0
+- Train Loss: 1.5656
+- Validation Loss: 1.7181
+- Train Accuracy: 0.8
 - Epoch: 4
 
 ## Model description
@@ -37,18 +37,18 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 55, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
+- optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 840, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
 - training_precision: float32
 
 ### Training results
 
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.4220     | 2.3051          | 0.3333         | 0     |
-| 2.3767     | 2.3182          | 0.3333         | 1     |
-| 2.3069     | 2.3403          | 0.0            | 2     |
-| 2.2764     | 2.3623          | 0.0            | 3     |
-| 2.3043     | 2.3868          | 0.0            | 4     |
+| 2.2701     | 2.2679          | 0.1            | 0     |
+| 2.1066     | 2.1493          | 0.4            | 1     |
+| 1.9323     | 2.0279          | 0.6            | 2     |
+| 1.7463     | 1.8713          | 0.6667         | 3     |
+| 1.5656     | 1.7181          | 0.8            | 4     |
 
 
 ### Framework versions

config.json

@@ -9,33 +9,31 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "boat",
-    "1": "children",
-    "10": "water",
-    "2": "dogs",
-    "3": "fireman",
-    "4": "firetruck",
-    "5": "mountains",
-    "6": "people",
-    "7": "river",
-    "8": "snow",
-    "9": "stairs"
+    "0": "AnnualCrop",
+    "1": "Forest",
+    "2": "HerbaceousVegetation",
+    "3": "Highway",
+    "4": "Industrial",
+    "5": "Pasture",
+    "6": "PermanentCrop",
+    "7": "Residential",
+    "8": "River",
+    "9": "SeaLake"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "boat": "0",
-    "children": "1",
-    "dogs": "2",
-    "fireman": "3",
-    "firetruck": "4",
-    "mountains": "5",
-    "people": "6",
-    "river": "7",
-    "snow": "8",
-    "stairs": "9",
-    "water": "10"
+    "AnnualCrop": "0",
+    "Forest": "1",
+    "HerbaceousVegetation": "2",
+    "Highway": "3",
+    "Industrial": "4",
+    "Pasture": "5",
+    "PermanentCrop": "6",
+    "Residential": "7",
+    "River": "8",
+    "SeaLake": "9"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9106ae9b3b3ffcb96078a9573b9aeb37b165399dcb8d43b1d7da8279ffd7b89c
-size 343513784
+oid sha256:31fc3058d526b7963ffda5c568157dc80935109fed134fd939d8a5a8df4a5e54
+size 343510712