Model save

README.md

@@ -0,0 +1,66 @@
+---
+license: apache-2.0
+base_model: EdBianchi/vit-fire-detection
+tags:
+- generated_from_trainer
+model-index:
+- name: fire_01
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# fire_01
+
+This model is a fine-tuned version of [EdBianchi/vit-fire-detection](https://huggingface.co/EdBianchi/vit-fire-detection) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.5913
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 2e-05
+- train_batch_size: 8
+- eval_batch_size: 8
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 10
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| No log        | 1.0   | 15   | 0.6945          |
+| No log        | 2.0   | 30   | 0.4927          |
+| No log        | 3.0   | 45   | 0.5516          |
+| No log        | 4.0   | 60   | 0.4700          |
+| No log        | 5.0   | 75   | 0.5426          |
+| No log        | 6.0   | 90   | 0.5149          |
+| No log        | 7.0   | 105  | 0.5401          |
+| No log        | 8.0   | 120  | 0.5532          |
+| No log        | 9.0   | 135  | 0.5662          |
+| No log        | 10.0  | 150  | 0.5913          |
+
+
+### Framework versions
+
+- Transformers 4.40.2
+- Pytorch 2.2.1+cu121
+- Datasets 2.19.1
+- Tokenizers 0.19.1

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4c9dae9fcd1270acf401e6f79c1591fc3b478e3ec27a844c42adba3378bb1b94
+oid sha256:6cea5e7da86fa2af45ad7b39b2d65dcaa6aad2b8310a9fffe8cd4ca9b5f61c1f
 size 343227052

runs/May16_15-11-14_3a89c76b2d21/events.out.tfevents.1715872275.3a89c76b2d21.187.2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7819420121afd75b94e57a77e9c1167cc231d48701d388d4ab1962c05e052ceb
-size 7150
+oid sha256:20a50de294b4311b7d9aa5b53762599d798aaa2f259a9c758519322761512bcf
+size 7775