Model save

README.md

@@ -1,7 +1,5 @@
 ---
 library_name: transformers
-license: apache-2.0
-base_model: google/vit-base-patch16-224-in21k
 tags:
 - generated_from_trainer
 metrics:
@@ -16,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # DF_Image_VIT_V1
 
-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.
+This model was trained from scratch on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0255
-- F1: 0.8152
+- Loss: 0.0889
+- F1: 0.5962
 
 ## Model description
 
@@ -44,17 +42,21 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 4
+- num_epochs: 8
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch | Step  | Validation Loss | F1     |
-|:-------------:|:-----:|:-----:|:---------------:|:------:|
-| 0.0462        | 1.0   | 4604  | 0.0570          | 0.4793 |
-| 0.0006        | 2.0   | 9208  | 0.0783          | 0.4261 |
-| 0.0006        | 3.0   | 13812 | 0.0492          | 0.6780 |
-| 0.0002        | 4.0   | 18416 | 0.0683          | 0.6523 |
+| Training Loss | Epoch | Step  | F1     | Validation Loss |
+|:-------------:|:-----:|:-----:|:------:|:---------------:|
+| 0.0462        | 1.0   | 4604  | 0.4793 | 0.0570          |
+| 0.0006        | 2.0   | 9208  | 0.4261 | 0.0783          |
+| 0.0006        | 3.0   | 13812 | 0.6780 | 0.0492          |
+| 0.0002        | 4.0   | 18416 | 0.6523 | 0.0683          |
+| 0.0001        | 5.0   | 23020 | 0.0833 | 0.5759          |
+| 0.0154        | 6.0   | 27624 | 0.0611 | 0.6762          |
+| 0.0001        | 7.0   | 32228 | 0.0836 | 0.6423          |
+| 0.0           | 8.0   | 36832 | 0.0889 | 0.5962          |
 
 
 ### Framework versions