Hugging Face's logo

rohan4s
/
finetuned-bangladeshi-traditional-food

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
rohan4s commited on
Commit
7005914
·
verified ·
1 Parent(s): ba865c0

Model save

Browse files
Files changed (5) hide show
  1. README.md +13 -7
  2. model.safetensors +1 -1
  3. runs/Dec09_09-16-02_2caa3c6c30c3/events.out.tfevents.1733735824.2caa3c6c30c3.23.0 +3 -0
  4. runs/Dec09_09-27-44_2caa3c6c30c3/events.out.tfevents.1733736480.2caa3c6c30c3.23.1 +3 -0
  5. training_args.bin +1 -1
README.md CHANGED
@@ -18,8 +18,8 @@ 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 an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.3410
22
- - Accuracy: 0.9218
23
 
24
  ## Model description
25
 
@@ -44,17 +44,23 @@ The following hyperparameters were used during training:
44
  - seed: 42
45
  - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
  - lr_scheduler_type: linear
47
- - num_epochs: 4
48
  - mixed_precision_training: Native AMP
49
 
50
  ### Training results
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
- | 1.0623 | 1.0 | 64 | 0.8796 | 0.8575 |
55
- | 0.5222 | 2.0 | 128 | 0.5890 | 0.8743 |
56
- | 0.3346 | 3.0 | 192 | 0.4126 | 0.9078 |
57
- | 0.2814 | 4.0 | 256 | 0.3410 | 0.9218 |
 
 
 
 
 
 
58
 
59
 
60
  ### 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 an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.7514
22
+ - Accuracy: 0.9162
23
 
24
  ## Model description
25
 
 
44
  - seed: 42
45
  - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
  - lr_scheduler_type: linear
47
+ - num_epochs: 10
48
  - mixed_precision_training: Native AMP
49
 
50
  ### Training results
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
+ | 0.2159 | 1.0 | 64 | 0.6425 | 0.8408 |
55
+ | 0.1968 | 2.0 | 128 | 0.6728 | 0.8547 |
56
+ | 0.0686 | 3.0 | 192 | 0.7183 | 0.8631 |
57
+ | 0.0975 | 4.0 | 256 | 0.8239 | 0.8464 |
58
+ | 0.0399 | 5.0 | 320 | 0.6338 | 0.8883 |
59
+ | 0.0221 | 6.0 | 384 | 0.7708 | 0.8883 |
60
+ | 0.0202 | 7.0 | 448 | 0.7649 | 0.8911 |
61
+ | 0.0274 | 8.0 | 512 | 0.7099 | 0.9162 |
62
+ | 0.0507 | 9.0 | 576 | 0.7499 | 0.9162 |
63
+ | 0.0653 | 10.0 | 640 | 0.7514 | 0.9162 |
64
 
65
 
66
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:563f29927af62c627e63597863b9e5070d7ea01cfedf80222ec6654b8e74d2ad
3
  size 343273192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:20981375782db6b197804bb0aba9451b6736e8e5370e7c980e2aa2284cd2d50c
3
  size 343273192
runs/Dec09_09-16-02_2caa3c6c30c3/events.out.tfevents.1733735824.2caa3c6c30c3.23.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f4fafdd5c5269c55a55a153bcc94785c0dcd9a452936cb8720e90cbe5bba010
3
+ size 27341
runs/Dec09_09-27-44_2caa3c6c30c3/events.out.tfevents.1733736480.2caa3c6c30c3.23.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31eeb057395f0b246845dd74cf286fd6b2134e8056e6b41f4d4583470017bf4c
3
+ size 23497
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1bc021ca16f6978199d06c83996671deba5cfbe528b7fc29f85b496ff808aedf
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2aa8832a3a36ea4a5f94642705207b1d66638f3d9a928a74f38c5a4aa05a60f1
3
  size 5304