End of training

README.md

@@ -23,16 +23,16 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.9253901789113057
+      value: 0.6404109589041096
     - name: F1
       type: f1
-      value: 0.9052377115229654
+      value: 0.5016949152542373
     - name: Precision
       type: precision
-      value: 0.9233171693926194
+      value: 0.6290224650880388
     - name: Recall
       type: recall
-      value: 0.8878526831581444
+      value: 0.41723721304873135
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -42,11 +42,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model was trained from scratch on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1768
-- Accuracy: 0.9254
-- F1: 0.9052
-- Precision: 0.9233
-- Recall: 0.8879
+- Loss: 0.7208
+- Accuracy: 0.6404
+- F1: 0.5017
+- Precision: 0.6290
+- Recall: 0.4172
 
 ## Model description
 
@@ -65,7 +65,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1e-05
+- learning_rate: 1e-06
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
@@ -74,17 +74,16 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 5
+- num_epochs: 4
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.3786        | 1.0   | 1970 | 0.3166          | 0.8590   | 0.8184 | 0.8469    | 0.7917 |
-| 0.2976        | 2.0   | 3941 | 0.2426          | 0.8952   | 0.8621 | 0.9138    | 0.8159 |
-| 0.2525        | 3.0   | 5911 | 0.2015          | 0.9144   | 0.8908 | 0.9132    | 0.8694 |
-| 0.2319        | 4.0   | 7882 | 0.1859          | 0.9216   | 0.9026 | 0.8996    | 0.9056 |
-| 0.206         | 5.0   | 9850 | 0.1768          | 0.9254   | 0.9052 | 0.9233    | 0.8879 |
+| 0.4992        | 1.0   | 2015 | 0.7072          | 0.6189   | 0.4517 | 0.6009    | 0.3619 |
+| 0.4581        | 2.0   | 4031 | 0.7145          | 0.6383   | 0.4787 | 0.6387    | 0.3828 |
+| 0.4229        | 3.0   | 6047 | 0.7146          | 0.6434   | 0.5077 | 0.6329    | 0.4238 |
+| 0.4096        | 4.0   | 8060 | 0.7208          | 0.6404   | 0.5017 | 0.6290    | 0.4172 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7eb1c18f6747ab7d9349841e9af93b1c7ccc2e52d733e956731a3cc7cc132ce9
+oid sha256:b49bc7310a3fc00621082b4cbcee85936597b66869eaa3582fa4e5f472ab5ae4
 size 110350128