pacojimon commited on
Commit
7483de3
1 Parent(s): ca9bc1c

Model save

Browse files
README.md ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: google/vit-base-patch16-224-in21k
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ - precision
9
+ - recall
10
+ - f1
11
+ model-index:
12
+ - name: finetuned-indian-food
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # finetuned-indian-food
20
+
21
+ 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.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.2293
24
+ - Accuracy: 0.9405
25
+ - Precision: 0.9395
26
+ - Recall: 0.9420
27
+ - F1: 0.9402
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 0.0002
47
+ - train_batch_size: 16
48
+ - eval_batch_size: 8
49
+ - seed: 42
50
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
+ - lr_scheduler_type: linear
52
+ - num_epochs: 4
53
+ - mixed_precision_training: Native AMP
54
+
55
+ ### Training results
56
+
57
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
58
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
59
+ | 0.8589 | 0.3 | 100 | 0.5618 | 0.8714 | 0.8981 | 0.8620 | 0.8696 |
60
+ | 0.6973 | 0.6 | 200 | 0.5544 | 0.8608 | 0.8742 | 0.8690 | 0.8630 |
61
+ | 0.4078 | 0.9 | 300 | 0.4671 | 0.8831 | 0.8915 | 0.8840 | 0.8812 |
62
+ | 0.3818 | 1.2 | 400 | 0.4203 | 0.8884 | 0.9017 | 0.8864 | 0.8877 |
63
+ | 0.2262 | 1.5 | 500 | 0.3481 | 0.9107 | 0.9177 | 0.9085 | 0.9098 |
64
+ | 0.2137 | 1.8 | 600 | 0.3761 | 0.9022 | 0.9094 | 0.9027 | 0.9026 |
65
+ | 0.4515 | 2.1 | 700 | 0.3722 | 0.9044 | 0.9091 | 0.9041 | 0.9017 |
66
+ | 0.3024 | 2.4 | 800 | 0.3105 | 0.9203 | 0.9198 | 0.9220 | 0.9188 |
67
+ | 0.1748 | 2.7 | 900 | 0.2767 | 0.9288 | 0.9274 | 0.9293 | 0.9272 |
68
+ | 0.1959 | 3.0 | 1000 | 0.2825 | 0.9256 | 0.9318 | 0.9243 | 0.9230 |
69
+ | 0.1663 | 3.3 | 1100 | 0.2549 | 0.9341 | 0.9362 | 0.9366 | 0.9356 |
70
+ | 0.0513 | 3.6 | 1200 | 0.2254 | 0.9416 | 0.9436 | 0.9422 | 0.9424 |
71
+ | 0.1478 | 3.9 | 1300 | 0.2293 | 0.9405 | 0.9395 | 0.9420 | 0.9402 |
72
+
73
+
74
+ ### Framework versions
75
+
76
+ - Transformers 4.39.3
77
+ - Pytorch 2.2.1+cu121
78
+ - Datasets 2.18.0
79
+ - Tokenizers 0.15.2
config.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "burger",
13
+ "1": "butter_naan",
14
+ "10": "kaathi_rolls",
15
+ "11": "kadai_paneer",
16
+ "12": "kulfi",
17
+ "13": "masala_dosa",
18
+ "14": "momos",
19
+ "15": "paani_puri",
20
+ "16": "pakode",
21
+ "17": "pav_bhaji",
22
+ "18": "pizza",
23
+ "19": "samosa",
24
+ "2": "chai",
25
+ "3": "chapati",
26
+ "4": "chole_bhature",
27
+ "5": "dal_makhani",
28
+ "6": "dhokla",
29
+ "7": "fried_rice",
30
+ "8": "idli",
31
+ "9": "jalebi"
32
+ },
33
+ "image_size": 224,
34
+ "initializer_range": 0.02,
35
+ "intermediate_size": 3072,
36
+ "label2id": {
37
+ "burger": "0",
38
+ "butter_naan": "1",
39
+ "chai": "2",
40
+ "chapati": "3",
41
+ "chole_bhature": "4",
42
+ "dal_makhani": "5",
43
+ "dhokla": "6",
44
+ "fried_rice": "7",
45
+ "idli": "8",
46
+ "jalebi": "9",
47
+ "kaathi_rolls": "10",
48
+ "kadai_paneer": "11",
49
+ "kulfi": "12",
50
+ "masala_dosa": "13",
51
+ "momos": "14",
52
+ "paani_puri": "15",
53
+ "pakode": "16",
54
+ "pav_bhaji": "17",
55
+ "pizza": "18",
56
+ "samosa": "19"
57
+ },
58
+ "layer_norm_eps": 1e-12,
59
+ "model_type": "vit",
60
+ "num_attention_heads": 12,
61
+ "num_channels": 3,
62
+ "num_hidden_layers": 12,
63
+ "patch_size": 16,
64
+ "problem_type": "single_label_classification",
65
+ "qkv_bias": true,
66
+ "torch_dtype": "float32",
67
+ "transformers_version": "4.39.3"
68
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5e836d3398a19623ddbd799d794037d33e6821c4a15c0b41602f1319b7a4e10
3
+ size 343279344
preprocessor_config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "do_resize",
5
+ "size",
6
+ "resample",
7
+ "do_rescale",
8
+ "rescale_factor",
9
+ "do_normalize",
10
+ "image_mean",
11
+ "image_std",
12
+ "return_tensors",
13
+ "data_format",
14
+ "input_data_format"
15
+ ],
16
+ "do_normalize": true,
17
+ "do_rescale": true,
18
+ "do_resize": true,
19
+ "image_mean": [
20
+ 0.5,
21
+ 0.5,
22
+ 0.5
23
+ ],
24
+ "image_processor_type": "ViTFeatureExtractor",
25
+ "image_std": [
26
+ 0.5,
27
+ 0.5,
28
+ 0.5
29
+ ],
30
+ "resample": 2,
31
+ "rescale_factor": 0.00392156862745098,
32
+ "size": {
33
+ "height": 224,
34
+ "width": 224
35
+ }
36
+ }
runs/Apr09_19-13-47_dcd6c694ee6d/events.out.tfevents.1712690184.dcd6c694ee6d.985.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0eeef3d7cd9286c5da3fa7967aa6b3f412bd09241570ac78718eb859b6e4092c
3
+ size 7030
runs/Apr09_19-13-47_dcd6c694ee6d/events.out.tfevents.1712690267.dcd6c694ee6d.985.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e4a0912840c4fb4605e65d047c624b93134de05cb77503dafac9450bedad46e
3
+ size 39919
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a81b835e5642bd802ff15acfcde5a7768463237f15df4a3180f27431970634cd
3
+ size 4920