Hugging Face's logo

YufeiWeng
/
donut-base-beans

Image Classification
Transformers
Safetensors
beit
vision
Generated from Trainer
Model card Files Community
YufeiWeng commited on
Commit
86c7ce2
·
verified ·
1 Parent(s): e2cbecc

Model save

Browse files
Files changed (2) hide show
  1. README.md +9 -12
  2. model.safetensors +1 -1
README.md CHANGED
@@ -2,8 +2,6 @@
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
- - image-classification
6
- - vision
7
  - generated_from_trainer
8
  metrics:
9
  - f1
@@ -19,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.0429
23
  - F1: 0.9705
24
 
25
  ## Model description
@@ -40,7 +38,7 @@ More information needed
40
 
41
  The following hyperparameters were used during training:
42
  - learning_rate: 3e-05
43
- - train_batch_size: 128
44
  - eval_batch_size: 64
45
  - seed: 1337
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
@@ -49,14 +47,13 @@ The following hyperparameters were used during training:
49
 
50
  ### Training results
51
 
52
- | Training Loss | Epoch | Step | F1 | Validation Loss |
53
- |:-------------:|:-----:|:-----:|:------:|:---------------:|
54
- | 0.0796 | 1.0 | 3367 | 0.4973 | 0.0687 |
55
- | 0.0611 | 2.0 | 6734 | 0.4973 | 0.0585 |
56
- | 0.0581 | 3.0 | 10101 | 0.4973 | 0.0591 |
57
- | 0.059 | 4.0 | 13468 | 0.4973 | 0.0550 |
58
- | 0.0563 | 5.0 | 16835 | 0.4973 | 0.0540 |
59
- | 0.0532 | 5.0 | 19805 | 0.9705 | 0.0429 |
60
 
61
 
62
  ### Framework versions
 
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
 
 
5
  - generated_from_trainer
6
  metrics:
7
  - f1
 
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.0446
21
  - F1: 0.9705
22
 
23
  ## Model description
 
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 3e-05
41
+ - train_batch_size: 64
42
  - eval_batch_size: 64
43
  - seed: 1337
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 
47
 
48
  ### Training results
49
 
50
+ | Training Loss | Epoch | Step | Validation Loss | F1 |
51
+ |:-------------:|:-----:|:-----:|:---------------:|:------:|
52
+ | 0.0659 | 1.0 | 3961 | 0.0543 | 0.9705 |
53
+ | 0.08 | 2.0 | 7922 | 0.0514 | 0.9705 |
54
+ | 0.0598 | 3.0 | 11883 | 0.0524 | 0.9705 |
55
+ | 0.0627 | 4.0 | 15844 | 0.0422 | 0.9705 |
56
+ | 0.0622 | 5.0 | 19805 | 0.0446 | 0.9705 |
 
57
 
58
 
59
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:20892a24822d391bb793f8afa633b6ce6109e7177fb57d3955ded77fdfe18c33
3
  size 343220892
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:500f0f73dc500410ebd16406eccf35a281a045e2314d6dd422b4a81a518a6a62
3
  size 343220892