Model save

README.md

@@ -4,11 +4,6 @@ license: other
 base_model: apple/mobilevit-xx-small
 tags:
 - generated_from_trainer
-metrics:
-- accuracy
-- precision
-- recall
-- f1
 model-index:
 - name: mobilevit-xx-small-FireRisk
   results: []
@@ -21,11 +16,16 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [apple/mobilevit-xx-small](https://huggingface.co/apple/mobilevit-xx-small) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.9969
-- Accuracy: 0.6119
-- Precision: 0.5205
-- Recall: 0.4804
-- F1: 0.4757
+- eval_loss: 0.9814
+- eval_accuracy: 0.6146
+- eval_precision: 0.5070
+- eval_recall: 0.4671
+- eval_f1: 0.4675
+- eval_runtime: 164.589
+- eval_samples_per_second: 130.878
+- eval_steps_per_second: 1.027
+- epoch: 1.0
+- step: 550
 
 ## Model description
 
@@ -52,17 +52,6 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 5
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
-| 1.1535        | 1.0   | 550  | 1.0990          | 0.5983   | 0.4529    | 0.4643 | 0.4518 |
-| 1.1244        | 2.0   | 1100 | 1.0341          | 0.6075   | 0.5178    | 0.4768 | 0.4656 |
-| 1.0729        | 3.0   | 1650 | 1.0131          | 0.6105   | 0.5131    | 0.4761 | 0.4686 |
-| 1.0514        | 4.0   | 2200 | 1.0073          | 0.6099   | 0.5149    | 0.4799 | 0.4761 |
-| 1.0221        | 5.0   | 2750 | 0.9969          | 0.6119   | 0.5205    | 0.4804 | 0.4757 |
-
-
 ### Framework versions
 
 - Transformers 4.45.1

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ef7a67435dd88bb79e7be53047be6e280a8f3aa71a2a23f6fe671535bbf65bc3
+oid sha256:fef0d38ce9e599d810858c1793ca44046fa272cb4a6eb502d7ddf406e4a2a5eb
 size 3875772