Hugging Face's logo

Shresthadev403
/
food-image-classification

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Model card Files
xet
 Metrics Community
Shresthadev403 commited on
Commit
e3a2a00
·
verified ·
1 Parent(s): e273453

End of training

Browse files
Files changed (8) hide show
  1. README.md +11 -8
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. runs/Feb04_07-25-37_0bc3247a45bd/events.out.tfevents.1707031537.0bc3247a45bd.819.4 +3 -0
  5. runs/Feb04_07-42-31_0bc3247a45bd/events.out.tfevents.1707032552.0bc3247a45bd.819.5 +3 -0
  6. runs/Feb04_07-48-42_0bc3247a45bd/events.out.tfevents.1707032923.0bc3247a45bd.819.6 +3 -0
  7. trainer_state.json +13 -28
  8. training_args.bin +1 -1
README.md CHANGED
@@ -1,4 +1,6 @@
1
  ---
 
 
2
  tags:
3
  - generated_from_trainer
4
  model-index:
@@ -11,15 +13,15 @@ should probably proofread and complete it, then remove this comment. -->
11
 
12
  # food-image-classification
13
 
14
- This model was trained from scratch on an unknown dataset.
15
  It achieves the following results on the evaluation set:
16
- - eval_loss: 1.5642
17
- - eval_accuracy: 0.896
18
- - eval_runtime: 15.4357
19
- - eval_samples_per_second: 64.785
20
- - eval_steps_per_second: 4.081
21
- - epoch: 0.32
22
- - step: 20
23
 
24
  ## Model description
25
 
@@ -48,6 +50,7 @@ The following hyperparameters were used during training:
48
  - lr_scheduler_type: linear
49
  - lr_scheduler_warmup_ratio: 0.1
50
  - num_epochs: 5
 
51
 
52
  ### Framework versions
53
 
 
1
  ---
2
+ license: apache-2.0
3
+ base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
  - generated_from_trainer
6
  model-index:
 
13
 
14
  # food-image-classification
15
 
16
+ 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.
17
  It achieves the following results on the evaluation set:
18
+ - eval_loss: 4.6215
19
+ - eval_accuracy: 0.0094
20
+ - eval_runtime: 140.7774
21
+ - eval_samples_per_second: 107.617
22
+ - eval_steps_per_second: 6.727
23
+ - epoch: 0.01
24
+ - step: 10
25
 
26
  ## Model description
27
 
 
50
  - lr_scheduler_type: linear
51
  - lr_scheduler_warmup_ratio: 0.1
52
  - num_epochs: 5
53
+ - mixed_precision_training: Native AMP
54
 
55
  ### Framework versions
56
 
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/content/food-image-classification",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
1
  {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7dd65022906a0ab5f82a88756f55cc79380928935ba4c8d864da8c5b57a3acff
3
  size 343528508
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:656c18fc5cba130c6fa5494776dee859077d1a75b06682410dbbdab4bc139c6c
3
  size 343528508
runs/Feb04_07-25-37_0bc3247a45bd/events.out.tfevents.1707031537.0bc3247a45bd.819.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d53e8ac1c0476273215c4ffe9bf269d110f3d655cbff853b316355639f029e0f
3
+ size 9228
runs/Feb04_07-42-31_0bc3247a45bd/events.out.tfevents.1707032552.0bc3247a45bd.819.5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69b31d75e5331ab79a98d2f2a9c2ee6210da2cd396afc921b98dc29ea7528019
3
+ size 9287
runs/Feb04_07-48-42_0bc3247a45bd/events.out.tfevents.1707032923.0bc3247a45bd.819.6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:25254f1aaefd9f6625f4d8c82f31a729e6d2d62e63f2a5b9fc79af946e941b6f
3
+ size 9623
trainer_state.json CHANGED
@@ -1,49 +1,34 @@
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
- "epoch": 0.32,
5
  "eval_steps": 10,
6
- "global_step": 20,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
- "epoch": 0.16,
13
- "learning_rate": 1.6129032258064517e-05,
14
- "loss": 1.5807,
15
  "step": 10
16
  },
17
  {
18
- "epoch": 0.16,
19
- "eval_accuracy": 0.889,
20
- "eval_loss": 1.6238535642623901,
21
- "eval_runtime": 15.4204,
22
- "eval_samples_per_second": 64.849,
23
- "eval_steps_per_second": 4.085,
24
  "step": 10
25
- },
26
- {
27
- "epoch": 0.32,
28
- "learning_rate": 3.2258064516129034e-05,
29
- "loss": 1.5408,
30
- "step": 20
31
- },
32
- {
33
- "epoch": 0.32,
34
- "eval_accuracy": 0.896,
35
- "eval_loss": 1.56419038772583,
36
- "eval_runtime": 15.4357,
37
- "eval_samples_per_second": 64.785,
38
- "eval_steps_per_second": 4.081,
39
- "step": 20
40
  }
41
  ],
42
  "logging_steps": 10,
43
- "max_steps": 310,
44
  "num_train_epochs": 5,
45
  "save_steps": 500,
46
- "total_flos": 9.9277758332928e+16,
47
  "trial_name": null,
48
  "trial_params": null
49
  }
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 0.010559662090813094,
5
  "eval_steps": 10,
6
+ "global_step": 10,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
+ "epoch": 0.01,
13
+ "learning_rate": 1.0548523206751055e-06,
14
+ "loss": 4.6245,
15
  "step": 10
16
  },
17
  {
18
+ "epoch": 0.01,
19
+ "eval_accuracy": 0.00943894389438944,
20
+ "eval_loss": 4.621488094329834,
21
+ "eval_runtime": 140.7774,
22
+ "eval_samples_per_second": 107.617,
23
+ "eval_steps_per_second": 6.727,
24
  "step": 10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  }
26
  ],
27
  "logging_steps": 10,
28
+ "max_steps": 4735,
29
  "num_train_epochs": 5,
30
  "save_steps": 500,
31
+ "total_flos": 4.9638879166464e+16,
32
  "trial_name": null,
33
  "trial_params": null
34
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:acef5c93ba342252b268b9013a40d50d329bbe50e1a0a135a0e8029a3284d5e8
3
  size 4600
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84434c2eca9b38915402890f1cae162ac37c75e9a0bb9cc160870a5a1bd9a643
3
  size 4600