Hugging Face's logo

MoGHenry
/
cat_dog_classifier

Image Classification
Transformers
Safetensors
resnet
Generated from Trainer
Inference Endpoints
Model card Files Community
MoGHenry commited on
Commit
82bb468
·
verified ·
1 Parent(s): d0d6430

End of training

Browse files
Files changed (3) hide show
  1. README.md +10 -10
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -7,19 +7,19 @@ tags:
7
  metrics:
8
  - accuracy
9
  model-index:
10
- - name: cat_dog_model
11
  results: []
12
  ---
13
 
14
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
  should probably proofread and complete it, then remove this comment. -->
16
 
17
- # cat_dog_model
18
 
19
  This model is a fine-tuned version of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0769
22
- - Accuracy: 0.9716
23
 
24
  ## Model description
25
 
@@ -39,7 +39,7 @@ More information needed
39
 
40
  The following hyperparameters were used during training:
41
  - learning_rate: 5e-05
42
- - train_batch_size: 32
43
  - eval_batch_size: 8
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
@@ -50,11 +50,11 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
- | 0.3965 | 1.0 | 586 | 0.1360 | 0.9560 |
54
- | 0.1441 | 2.0 | 1172 | 0.0976 | 0.9624 |
55
- | 0.1113 | 3.0 | 1758 | 0.0805 | 0.9692 |
56
- | 0.1061 | 4.0 | 2344 | 0.0844 | 0.9673 |
57
- | 0.1002 | 5.0 | 2930 | 0.0769 | 0.9716 |
58
 
59
 
60
  ### Framework versions
 
7
  metrics:
8
  - accuracy
9
  model-index:
10
+ - name: cat_dog_classifier
11
  results: []
12
  ---
13
 
14
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
  should probably proofread and complete it, then remove this comment. -->
16
 
17
+ # cat_dog_classifier
18
 
19
  This model is a fine-tuned version of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0884
22
+ - Accuracy: 0.9688
23
 
24
  ## Model description
25
 
 
39
 
40
  The following hyperparameters were used during training:
41
  - learning_rate: 5e-05
42
+ - train_batch_size: 64
43
  - eval_batch_size: 8
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | No log | 1.0 | 293 | 0.1872 | 0.9440 |
54
+ | 0.3326 | 2.0 | 586 | 0.1120 | 0.9590 |
55
+ | 0.3326 | 3.0 | 879 | 0.0877 | 0.9677 |
56
+ | 0.1182 | 4.0 | 1172 | 0.0940 | 0.9641 |
57
+ | 0.1182 | 5.0 | 1465 | 0.0884 | 0.9688 |
58
 
59
 
60
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:610fd3d2b12311f196ab6744f32c9911089cdb4cea0d64eae26a85434ae171ca
3
  size 94302952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:add5a98fcab1d4d20024106d55326ced0dc235b53482385625c0d1e0a0b40347
3
  size 94302952
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:da7dfc7ccdf5656398950e4149420cd405c66bfdbc999725073fcd6bc6d187ed
3
  size 5112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83e8a813ed98b4a7d74e035352d2163af7ce6918bdf07887dc2b6b1b8606ac38
3
  size 5112