Kushagra07 commited on
Commit
f9dfdf2
1 Parent(s): 6da50b5

End of training

Browse files
README.md ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: google/vit-base-patch16-224
4
+ tags:
5
+ - generated_from_trainer
6
+ datasets:
7
+ - imagefolder
8
+ metrics:
9
+ - accuracy
10
+ - recall
11
+ - f1
12
+ - precision
13
+ model-index:
14
+ - name: vit-base-patch16-224-finetuned-ind-17-imbalanced-aadhaarmask
15
+ results:
16
+ - task:
17
+ name: Image Classification
18
+ type: image-classification
19
+ dataset:
20
+ name: imagefolder
21
+ type: imagefolder
22
+ config: default
23
+ split: train
24
+ args: default
25
+ metrics:
26
+ - name: Accuracy
27
+ type: accuracy
28
+ value: 0.848446147296722
29
+ - name: Recall
30
+ type: recall
31
+ value: 0.848446147296722
32
+ - name: F1
33
+ type: f1
34
+ value: 0.8477849036950597
35
+ - name: Precision
36
+ type: precision
37
+ value: 0.8513434130555053
38
+ ---
39
+
40
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
41
+ should probably proofread and complete it, then remove this comment. -->
42
+
43
+ # vit-base-patch16-224-finetuned-ind-17-imbalanced-aadhaarmask
44
+
45
+ This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
46
+ It achieves the following results on the evaluation set:
47
+ - Loss: 0.3494
48
+ - Accuracy: 0.8484
49
+ - Recall: 0.8484
50
+ - F1: 0.8478
51
+ - Precision: 0.8513
52
+
53
+ ## Model description
54
+
55
+ More information needed
56
+
57
+ ## Intended uses & limitations
58
+
59
+ More information needed
60
+
61
+ ## Training and evaluation data
62
+
63
+ More information needed
64
+
65
+ ## Training procedure
66
+
67
+ ### Training hyperparameters
68
+
69
+ The following hyperparameters were used during training:
70
+ - learning_rate: 5e-05
71
+ - train_batch_size: 8
72
+ - eval_batch_size: 8
73
+ - seed: 42
74
+ - gradient_accumulation_steps: 4
75
+ - total_train_batch_size: 32
76
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
77
+ - lr_scheduler_type: linear
78
+ - lr_scheduler_warmup_ratio: 0.1
79
+ - num_epochs: 10
80
+
81
+ ### Training results
82
+
83
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Recall | F1 | Precision |
84
+ |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:------:|:---------:|
85
+ | 0.5792 | 0.9974 | 293 | 0.5989 | 0.7969 | 0.7969 | 0.7829 | 0.7897 |
86
+ | 0.42 | 1.9983 | 587 | 0.5251 | 0.8046 | 0.8046 | 0.7960 | 0.7985 |
87
+ | 0.3501 | 2.9991 | 881 | 0.4299 | 0.8335 | 0.8335 | 0.8312 | 0.8363 |
88
+ | 0.3187 | 4.0 | 1175 | 0.4302 | 0.8169 | 0.8169 | 0.8144 | 0.8182 |
89
+ | 0.3873 | 4.9974 | 1468 | 0.4246 | 0.8250 | 0.8250 | 0.8238 | 0.8326 |
90
+ | 0.3786 | 5.9983 | 1762 | 0.3881 | 0.8306 | 0.8306 | 0.8303 | 0.8394 |
91
+ | 0.337 | 6.9991 | 2056 | 0.3803 | 0.8306 | 0.8306 | 0.8304 | 0.8351 |
92
+ | 0.2717 | 8.0 | 2350 | 0.3785 | 0.8395 | 0.8395 | 0.8361 | 0.8482 |
93
+ | 0.2753 | 8.9974 | 2643 | 0.3805 | 0.8327 | 0.8327 | 0.8314 | 0.8346 |
94
+ | 0.2814 | 9.9745 | 2930 | 0.3362 | 0.8480 | 0.8480 | 0.8467 | 0.8499 |
95
+
96
+
97
+ ### Framework versions
98
+
99
+ - Transformers 4.40.1
100
+ - Pytorch 2.2.0a0+81ea7a4
101
+ - Datasets 2.19.0
102
+ - Tokenizers 0.19.1
emissions.csv ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ timestamp,project_name,run_id,duration,emissions,emissions_rate,cpu_power,gpu_power,ram_power,cpu_energy,gpu_energy,ram_energy,energy_consumed,country_name,country_iso_code,region,cloud_provider,cloud_region,os,python_version,codecarbon_version,cpu_count,cpu_model,gpu_count,gpu_model,longitude,latitude,ram_total_size,tracking_mode,on_cloud,pue
2
+ 2024-05-02T04:16:46,codecarbon,d0fbfb8d-c22b-48bf-83a8-55f401614597,1385.1880159378052,0.00014365478459774659,1.0370778763956378e-07,42.5,74.16619667871524,11.667008399963379,0.016352169314192398,0.03959190806238999,0.004485696278541486,0.060429773655123926,Canada,CAN,quebec,,,Linux-5.15.0-105-generic-x86_64-with-glibc2.35,3.10.12,2.3.5,32,13th Gen Intel(R) Core(TM) i9-13900K,1,1 x NVIDIA GeForce RTX 4060 Ti,-71.2,46.8,31.112022399902344,machine,N,1.0
runs/May02_03-53-36_60f4804cf903/events.out.tfevents.1714623470.60f4804cf903.2810.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df272ad4ffe8e5be3cc99d275da67a39801e867cd42a1d8a1541a2d09492836e
3
+ size 560