AshutoshD245 commited on
Commit
a8c72b9
1 Parent(s): 1890e37

End of training

Browse files
Files changed (2) hide show
  1. README.md +10 -10
  2. tf_model.h5 +2 -2
README.md CHANGED
@@ -4,20 +4,20 @@ base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
  - generated_from_keras_callback
6
  model-index:
7
- - name: AIUser245/food_classifier
8
  results: []
9
  ---
10
 
11
  <!-- This model card has been generated automatically according to the information Keras had access to. You should
12
  probably proofread and complete it, then remove this comment. -->
13
 
14
- # AIUser245/food_classifier
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
- - Train Loss: 0.3703
19
- - Validation Loss: 0.3299
20
- - Train Accuracy: 0.924
21
  - Epoch: 4
22
 
23
  ## Model description
@@ -44,11 +44,11 @@ The following hyperparameters were used during training:
44
 
45
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
46
  |:----------:|:---------------:|:--------------:|:-----:|
47
- | 2.7963 | 1.5950 | 0.861 | 0 |
48
- | 1.2101 | 0.8091 | 0.886 | 1 |
49
- | 0.6770 | 0.4879 | 0.922 | 2 |
50
- | 0.5020 | 0.3645 | 0.927 | 3 |
51
- | 0.3703 | 0.3299 | 0.924 | 4 |
52
 
53
 
54
  ### Framework versions
 
4
  tags:
5
  - generated_from_keras_callback
6
  model-index:
7
+ - name: AshutoshD245/food_classifier
8
  results: []
9
  ---
10
 
11
  <!-- This model card has been generated automatically according to the information Keras had access to. You should
12
  probably proofread and complete it, then remove this comment. -->
13
 
14
+ # AshutoshD245/food_classifier
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
+ - Train Loss: 0.3889
19
+ - Validation Loss: 0.3585
20
+ - Train Accuracy: 0.914
21
  - Epoch: 4
22
 
23
  ## Model description
 
44
 
45
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
46
  |:----------:|:---------------:|:--------------:|:-----:|
47
+ | 2.8233 | 1.6956 | 0.808 | 0 |
48
+ | 1.2230 | 0.8527 | 0.882 | 1 |
49
+ | 0.7043 | 0.5496 | 0.896 | 2 |
50
+ | 0.4912 | 0.4837 | 0.882 | 3 |
51
+ | 0.3889 | 0.3585 | 0.914 | 4 |
52
 
53
 
54
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:718953e62b8a42345344f9e8df146ece355ef8bf13d5c356d4b5a16a7f737edf
3
- size 343790264
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7589af06097cfd17ad12067c7f32a4b77c795e1601195605bcba1b8663fd837
3
+ size 343773880