Hugging Face's logo

bhargob11
/
vit-base-patch16-224-in21k-finetuned-housplants

TensorBoard
Safetensors
vit
Generated from Trainer
Eval Results
Model card Files Metrics Community
bhargob11 commited on
Commit
b220578
·
verified ·
1 Parent(s): 1c65c65

End of training

Browse files
Files changed (6) hide show
  1. README.md +1 -1
  2. all_results.json +13 -0
  3. eval_results.json +8 -0
  4. runs/Aug15_05-59-29_0ddf4ef5dd42/events.out.tfevents.1723701858.0ddf4ef5dd42.2650.1 +3 -0
  5. train_results.json +8 -0
  6. trainer_state.json +108 -0
README.md CHANGED
@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.3899
36
  - Accuracy: 0.8673
37
 
38
  ## Model description
 
32
 
33
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.4007
36
  - Accuracy: 0.8673
37
 
38
  ## Model description
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_accuracy": 0.8673469387755102,
4
+ "eval_loss": 0.4006994068622589,
5
+ "eval_runtime": 2.3939,
6
+ "eval_samples_per_second": 40.938,
7
+ "eval_steps_per_second": 1.671,
8
+ "total_flos": 3.382525346676941e+17,
9
+ "train_loss": 0.4982973950249808,
10
+ "train_runtime": 161.5513,
11
+ "train_samples_per_second": 27.019,
12
+ "train_steps_per_second": 0.217
13
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_accuracy": 0.8673469387755102,
4
+ "eval_loss": 0.4006994068622589,
5
+ "eval_runtime": 2.3939,
6
+ "eval_samples_per_second": 40.938,
7
+ "eval_steps_per_second": 1.671
8
+ }
runs/Aug15_05-59-29_0ddf4ef5dd42/events.out.tfevents.1723701858.0ddf4ef5dd42.2650.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:976c2b041892fba6386d7478ecf8b9a37980d28d6ff39fff8bf4c04a052d58bf
3
+ size 405
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "total_flos": 3.382525346676941e+17,
4
+ "train_loss": 0.4982973950249808,
5
+ "train_runtime": 161.5513,
6
+ "train_samples_per_second": 27.019,
7
+ "train_steps_per_second": 0.217
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.8673469387755102,
3
+ "best_model_checkpoint": "vit-base-patch16-224-in21k-finetuned-housplants/checkpoint-28",
4
+ "epoch": 5.0,
5
+ "eval_steps": 500,
6
+ "global_step": 35,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 1.0,
13
+ "eval_accuracy": 0.7346938775510204,
14
+ "eval_loss": 0.619100034236908,
15
+ "eval_runtime": 2.3859,
16
+ "eval_samples_per_second": 41.075,
17
+ "eval_steps_per_second": 1.677,
18
+ "step": 7
19
+ },
20
+ {
21
+ "epoch": 1.4285714285714286,
22
+ "grad_norm": 0.6459439992904663,
23
+ "learning_rate": 4.032258064516129e-05,
24
+ "loss": 0.663,
25
+ "step": 10
26
+ },
27
+ {
28
+ "epoch": 2.0,
29
+ "eval_accuracy": 0.8571428571428571,
30
+ "eval_loss": 0.4970492124557495,
31
+ "eval_runtime": 2.3776,
32
+ "eval_samples_per_second": 41.219,
33
+ "eval_steps_per_second": 1.682,
34
+ "step": 14
35
+ },
36
+ {
37
+ "epoch": 2.857142857142857,
38
+ "grad_norm": 0.7293534874916077,
39
+ "learning_rate": 2.4193548387096777e-05,
40
+ "loss": 0.5043,
41
+ "step": 20
42
+ },
43
+ {
44
+ "epoch": 3.0,
45
+ "eval_accuracy": 0.8571428571428571,
46
+ "eval_loss": 0.4364696741104126,
47
+ "eval_runtime": 2.38,
48
+ "eval_samples_per_second": 41.176,
49
+ "eval_steps_per_second": 1.681,
50
+ "step": 21
51
+ },
52
+ {
53
+ "epoch": 4.0,
54
+ "eval_accuracy": 0.8673469387755102,
55
+ "eval_loss": 0.4006994068622589,
56
+ "eval_runtime": 2.3915,
57
+ "eval_samples_per_second": 40.978,
58
+ "eval_steps_per_second": 1.673,
59
+ "step": 28
60
+ },
61
+ {
62
+ "epoch": 4.285714285714286,
63
+ "grad_norm": 0.7392838597297668,
64
+ "learning_rate": 8.064516129032258e-06,
65
+ "loss": 0.3979,
66
+ "step": 30
67
+ },
68
+ {
69
+ "epoch": 5.0,
70
+ "eval_accuracy": 0.8673469387755102,
71
+ "eval_loss": 0.38987594842910767,
72
+ "eval_runtime": 2.4565,
73
+ "eval_samples_per_second": 39.894,
74
+ "eval_steps_per_second": 1.628,
75
+ "step": 35
76
+ },
77
+ {
78
+ "epoch": 5.0,
79
+ "step": 35,
80
+ "total_flos": 3.382525346676941e+17,
81
+ "train_loss": 0.4982973950249808,
82
+ "train_runtime": 161.5513,
83
+ "train_samples_per_second": 27.019,
84
+ "train_steps_per_second": 0.217
85
+ }
86
+ ],
87
+ "logging_steps": 10,
88
+ "max_steps": 35,
89
+ "num_input_tokens_seen": 0,
90
+ "num_train_epochs": 5,
91
+ "save_steps": 500,
92
+ "stateful_callbacks": {
93
+ "TrainerControl": {
94
+ "args": {
95
+ "should_epoch_stop": false,
96
+ "should_evaluate": false,
97
+ "should_log": false,
98
+ "should_save": true,
99
+ "should_training_stop": true
100
+ },
101
+ "attributes": {}
102
+ }
103
+ },
104
+ "total_flos": 3.382525346676941e+17,
105
+ "train_batch_size": 32,
106
+ "trial_name": null,
107
+ "trial_params": null
108
+ }