prakhardixit24 commited on
Commit
9e08622
·
verified ·
1 Parent(s): ad02e33

End of training

Browse files
Files changed (1) hide show
  1. README.md +7 -7
README.md CHANGED
@@ -17,12 +17,12 @@ model-index:
17
  name: imagefolder
18
  type: imagefolder
19
  config: default
20
- split: train[:18]
21
  args: default
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
- value: 0.75
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ 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.6055
36
- - Accuracy: 0.75
37
 
38
  ## Model description
39
 
@@ -67,9 +67,9 @@ The following hyperparameters were used during training:
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
- | No log | 1.0 | 1 | 0.7063 | 0.25 |
71
- | No log | 2.0 | 2 | 0.6390 | 0.75 |
72
- | No log | 3.0 | 3 | 0.6055 | 0.75 |
73
 
74
 
75
  ### Framework versions
 
17
  name: imagefolder
18
  type: imagefolder
19
  config: default
20
+ split: train[:39]
21
  args: default
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
+ value: 0.625
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
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.6614
36
+ - Accuracy: 0.625
37
 
38
  ## Model description
39
 
 
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
+ | No log | 1.0 | 1 | 0.7041 | 0.625 |
71
+ | No log | 2.0 | 2 | 0.6657 | 0.625 |
72
+ | No log | 3.0 | 3 | 0.6614 | 0.625 |
73
 
74
 
75
  ### Framework versions