Hugging Face's logo

heado
/
ViT_beans

Image Classification
Transformers
TensorBoard
Safetensors
vit
HHD
3_class
ViT
Generated from Trainer
Model card Files
xet
Metrics Community
heado commited on
Commit
dfaf301
·
verified ·
1 Parent(s): 93f3bcb

heado

Browse files
Files changed (7) hide show
  1. README.md +9 -10
  2. model.safetensors +1 -1
  3. runs/Sep02_03-54-25_3d244b9d4949/events.out.tfevents.1725249281.3d244b9d4949.6417.11 +3 -0
  4. runs/Sep02_03-55-00_3d244b9d4949/events.out.tfevents.1725249308.3d244b9d4949.6417.12 +3 -0
  5. runs/Sep02_03-55-37_3d244b9d4949/events.out.tfevents.1725249343.3d244b9d4949.6417.13 +3 -0
  6. runs/Sep02_03-59-20_3d244b9d4949/events.out.tfevents.1725249566.3d244b9d4949.6417.14 +3 -0
  7. training_args.bin +1 -1
README.md CHANGED
@@ -18,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  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 beans dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0515
22
 
23
  ## Model description
24
 
@@ -37,21 +37,20 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - learning_rate: 0.0002
41
- - train_batch_size: 16
42
- - eval_batch_size: 8
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
- - num_epochs: 4
47
- - mixed_precision_training: Native AMP
48
 
49
  ### Training results
50
 
51
- | Training Loss | Epoch | Step | Validation Loss |
52
- |:-------------:|:------:|:----:|:---------------:|
53
- | 0.0722 | 1.5385 | 100 | 0.1296 |
54
- | 0.0629 | 3.0769 | 200 | 0.0515 |
55
 
56
 
57
  ### Framework versions
 
18
 
19
  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 beans dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.7118
22
 
23
  ## Model description
24
 
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 2e-05
41
+ - train_batch_size: 64
42
+ - eval_batch_size: 64
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
+ - num_epochs: 2
 
47
 
48
  ### Training results
49
 
50
+ | Training Loss | Epoch | Step | Validation Loss |
51
+ |:-------------:|:-----:|:----:|:---------------:|
52
+ | No log | 1.0 | 17 | 0.8068 |
53
+ | No log | 2.0 | 34 | 0.7118 |
54
 
55
 
56
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:33c98fe4f5721871e37977cb56cce201f5178926cf7e596ad70ade08567f9a38
3
  size 343227052
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dda850322656ad21db7ee4f3cdb2526caafdc3d823ab7057373c6881d8ed6c54
3
  size 343227052
runs/Sep02_03-54-25_3d244b9d4949/events.out.tfevents.1725249281.3d244b9d4949.6417.11 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d39cf444f3e10341f15f1386fbdc2296107b43333fc25479d1b66487bf68a90
3
+ size 4976
runs/Sep02_03-55-00_3d244b9d4949/events.out.tfevents.1725249308.3d244b9d4949.6417.12 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f5a9cc381c0858a736f353b796bffd4b6e6442bb62dd57ebf770b8128800807
3
+ size 4975
runs/Sep02_03-55-37_3d244b9d4949/events.out.tfevents.1725249343.3d244b9d4949.6417.13 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99148d2b95f6719ab9a24a6ff0b230fd77de946186f86565233d74e725c5b76f
3
+ size 4977
runs/Sep02_03-59-20_3d244b9d4949/events.out.tfevents.1725249566.3d244b9d4949.6417.14 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e582513d4d77925f3341e76bab1accbed8c6389ddeb946720a9acc2c695cefc5
3
+ size 5763
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f86f8444eab9c95536c114323caefb5df5c77ab4a1d2e5b600052ec7ed6968b0
3
  size 5112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fea6ad73f1c81dd6e8ab8d64b356b8ccc5c7eef8a872625890e1fdf5adffd3c
3
  size 5112