SachaEL commited on
Commit
3664e0b
1 Parent(s): a6ffdab

Model save

Browse files
README.md ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: EdBianchi/vit-fire-detection
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: fire_01
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # fire_01
15
+
16
+ This model is a fine-tuned version of [EdBianchi/vit-fire-detection](https://huggingface.co/EdBianchi/vit-fire-detection) on the None dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 0.5913
19
+
20
+ ## Model description
21
+
22
+ More information needed
23
+
24
+ ## Intended uses & limitations
25
+
26
+ More information needed
27
+
28
+ ## Training and evaluation data
29
+
30
+ More information needed
31
+
32
+ ## Training procedure
33
+
34
+ ### Training hyperparameters
35
+
36
+ The following hyperparameters were used during training:
37
+ - learning_rate: 2e-05
38
+ - train_batch_size: 8
39
+ - eval_batch_size: 8
40
+ - seed: 42
41
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
+ - lr_scheduler_type: linear
43
+ - num_epochs: 10
44
+
45
+ ### Training results
46
+
47
+ | Training Loss | Epoch | Step | Validation Loss |
48
+ |:-------------:|:-----:|:----:|:---------------:|
49
+ | No log | 1.0 | 15 | 0.6945 |
50
+ | No log | 2.0 | 30 | 0.4927 |
51
+ | No log | 3.0 | 45 | 0.5516 |
52
+ | No log | 4.0 | 60 | 0.4700 |
53
+ | No log | 5.0 | 75 | 0.5426 |
54
+ | No log | 6.0 | 90 | 0.5149 |
55
+ | No log | 7.0 | 105 | 0.5401 |
56
+ | No log | 8.0 | 120 | 0.5532 |
57
+ | No log | 9.0 | 135 | 0.5662 |
58
+ | No log | 10.0 | 150 | 0.5913 |
59
+
60
+
61
+ ### Framework versions
62
+
63
+ - Transformers 4.40.2
64
+ - Pytorch 2.2.1+cu121
65
+ - Datasets 2.19.1
66
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4c9dae9fcd1270acf401e6f79c1591fc3b478e3ec27a844c42adba3378bb1b94
3
  size 343227052
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cea5e7da86fa2af45ad7b39b2d65dcaa6aad2b8310a9fffe8cd4ca9b5f61c1f
3
  size 343227052
runs/May16_15-11-14_3a89c76b2d21/events.out.tfevents.1715872275.3a89c76b2d21.187.2 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7819420121afd75b94e57a77e9c1167cc231d48701d388d4ab1962c05e052ceb
3
- size 7150
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:20a50de294b4311b7d9aa5b53762599d798aaa2f259a9c758519322761512bcf
3
+ size 7775