mhdiqbalpradipta commited on
Commit
6a915e0
1 Parent(s): 0843436

End of training

Browse files
README.md CHANGED
@@ -5,9 +5,24 @@ tags:
5
  - generated_from_trainer
6
  datasets:
7
  - cats_vs_dogs
 
 
8
  model-index:
9
  - name: cat_or_dogs
10
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ---
12
 
13
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,6 +31,9 @@ should probably proofread and complete it, then remove this comment. -->
16
  # cat_or_dogs
17
 
18
  This model is a fine-tuned version of [akahana/vit-base-cats-vs-dogs](https://huggingface.co/akahana/vit-base-cats-vs-dogs) on the cats_vs_dogs dataset.
 
 
 
19
 
20
  ## Model description
21
 
@@ -43,6 +61,15 @@ The following hyperparameters were used during training:
43
  - num_epochs: 3
44
  - mixed_precision_training: Native AMP
45
 
 
 
 
 
 
 
 
 
 
46
  ### Framework versions
47
 
48
  - Transformers 4.37.2
 
5
  - generated_from_trainer
6
  datasets:
7
  - cats_vs_dogs
8
+ metrics:
9
+ - accuracy
10
  model-index:
11
  - name: cat_or_dogs
12
+ results:
13
+ - task:
14
+ name: Image Classification
15
+ type: image-classification
16
+ dataset:
17
+ name: cats_vs_dogs
18
+ type: cats_vs_dogs
19
+ config: default
20
+ split: train
21
+ args: default
22
+ metrics:
23
+ - name: Accuracy
24
+ type: accuracy
25
+ value: 0.9820589491670226
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
  # cat_or_dogs
32
 
33
  This model is a fine-tuned version of [akahana/vit-base-cats-vs-dogs](https://huggingface.co/akahana/vit-base-cats-vs-dogs) on the cats_vs_dogs dataset.
34
+ It achieves the following results on the evaluation set:
35
+ - Loss: 0.0561
36
+ - Accuracy: 0.9821
37
 
38
  ## Model description
39
 
 
61
  - num_epochs: 3
62
  - mixed_precision_training: Native AMP
63
 
64
+ ### Training results
65
+
66
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 0.0389 | 1.0 | 1171 | 0.0638 | 0.9793 |
69
+ | 0.0682 | 2.0 | 2342 | 0.0510 | 0.9812 |
70
+ | 0.0623 | 3.0 | 3513 | 0.0561 | 0.9821 |
71
+
72
+
73
  ### Framework versions
74
 
75
  - Transformers 4.37.2
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0159f263d95c7912c530551fdf9bfc7d3c8a9b7f866077781d4014e8d9baf1f2
3
  size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d3e465116709bcdbe292b6b4e1a5e52626d5161a403462b9f4a199353196a6f
3
  size 343223968
runs/Feb24_10-11-37_73a7ae1ecc27/events.out.tfevents.1708769498.73a7ae1ecc27.220.3 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4fb203359f75a481eee5e8a93a276bf765ddfb8261bd367391ade8db4289fee8
3
- size 4997
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d89a82ae10e14d1cd58343295797a06f9b62f3b79f86d4d873710c348ae7693a
3
+ size 6311