Commit
•
f7b64b1
1
Parent(s):
b701a34
Training in progress epoch 0
Browse files
README.md
CHANGED
@@ -15,11 +15,11 @@ probably proofread and complete it, then remove this comment. -->
|
|
15 |
|
16 |
This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
|
17 |
It achieves the following results on the evaluation set:
|
18 |
-
- Train Loss: 0.
|
19 |
-
- Train Accuracy: 0.
|
20 |
-
- Train Top-3-accuracy: 0.
|
21 |
-
- Validation Loss: 0.
|
22 |
-
- Validation Accuracy:
|
23 |
- Validation Top-3-accuracy: 1.0
|
24 |
- Epoch: 0
|
25 |
|
@@ -40,19 +40,19 @@ More information needed
|
|
40 |
### Training hyperparameters
|
41 |
|
42 |
The following hyperparameters were used during training:
|
43 |
-
- optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate':
|
44 |
- training_precision: mixed_float16
|
45 |
|
46 |
### Training results
|
47 |
|
48 |
| Train Loss | Train Accuracy | Train Top-3-accuracy | Validation Loss | Validation Accuracy | Validation Top-3-accuracy | Epoch |
|
49 |
|:----------:|:--------------:|:--------------------:|:---------------:|:-------------------:|:-------------------------:|:-----:|
|
50 |
-
| 0.
|
51 |
|
52 |
|
53 |
### Framework versions
|
54 |
|
55 |
- Transformers 4.31.0
|
56 |
- TensorFlow 2.12.0
|
57 |
-
- Datasets 2.
|
58 |
- Tokenizers 0.13.3
|
|
|
15 |
|
16 |
This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
|
17 |
It achieves the following results on the evaluation set:
|
18 |
+
- Train Loss: 0.1415
|
19 |
+
- Train Accuracy: 0.9739
|
20 |
+
- Train Top-3-accuracy: 0.9863
|
21 |
+
- Validation Loss: 0.0055
|
22 |
+
- Validation Accuracy: 0.9993
|
23 |
- Validation Top-3-accuracy: 1.0
|
24 |
- Epoch: 0
|
25 |
|
|
|
40 |
### Training hyperparameters
|
41 |
|
42 |
The following hyperparameters were used during training:
|
43 |
+
- optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 0.0002, 'decay_steps': 2720, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
|
44 |
- training_precision: mixed_float16
|
45 |
|
46 |
### Training results
|
47 |
|
48 |
| Train Loss | Train Accuracy | Train Top-3-accuracy | Validation Loss | Validation Accuracy | Validation Top-3-accuracy | Epoch |
|
49 |
|:----------:|:--------------:|:--------------------:|:---------------:|:-------------------:|:-------------------------:|:-----:|
|
50 |
+
| 0.1415 | 0.9739 | 0.9863 | 0.0055 | 0.9993 | 1.0 | 0 |
|
51 |
|
52 |
|
53 |
### Framework versions
|
54 |
|
55 |
- Transformers 4.31.0
|
56 |
- TensorFlow 2.12.0
|
57 |
+
- Datasets 2.14.0
|
58 |
- Tokenizers 0.13.3
|
config.json
CHANGED
@@ -9,37 +9,113 @@
|
|
9 |
"hidden_dropout_prob": 0.0,
|
10 |
"hidden_size": 768,
|
11 |
"id2label": {
|
12 |
-
"0": "
|
13 |
-
"1": "
|
14 |
"10": "cereal_box",
|
15 |
-
"11": "
|
16 |
-
"12": "
|
17 |
-
"
|
18 |
-
"
|
19 |
-
"
|
20 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
"6": "calculator",
|
22 |
-
"7": "
|
23 |
-
"8": "
|
24 |
-
"9": "
|
25 |
},
|
26 |
"image_size": 224,
|
27 |
"initializer_range": 0.02,
|
28 |
"intermediate_size": 3072,
|
29 |
"label2id": {
|
30 |
-
"
|
31 |
-
"
|
32 |
-
"
|
33 |
-
"
|
34 |
-
"
|
|
|
35 |
"calculator": "6",
|
36 |
-
"camera": "
|
37 |
-
"cap": "
|
38 |
-
"cell_phone": "
|
39 |
"cereal_box": "10",
|
40 |
-
"coffee_mug": "
|
41 |
-
"comb": "
|
42 |
-
"dry_battery": "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
43 |
},
|
44 |
"layer_norm_eps": 1e-12,
|
45 |
"model_type": "vit",
|
|
|
9 |
"hidden_dropout_prob": 0.0,
|
10 |
"hidden_size": 768,
|
11 |
"id2label": {
|
12 |
+
"0": "apple",
|
13 |
+
"1": "ball",
|
14 |
"10": "cereal_box",
|
15 |
+
"11": "coffee_mug",
|
16 |
+
"12": "comb",
|
17 |
+
"13": "dry_battery",
|
18 |
+
"14": "flashlight",
|
19 |
+
"15": "food_bag",
|
20 |
+
"16": "food_box",
|
21 |
+
"17": "food_can",
|
22 |
+
"18": "food_cup",
|
23 |
+
"19": "food_jar",
|
24 |
+
"2": "banana",
|
25 |
+
"20": "garlic",
|
26 |
+
"21": "glue_stick",
|
27 |
+
"22": "greens",
|
28 |
+
"23": "hand_towel",
|
29 |
+
"24": "instant_noodles",
|
30 |
+
"25": "keyboard",
|
31 |
+
"26": "kleenex",
|
32 |
+
"27": "lemon",
|
33 |
+
"28": "lightbulb",
|
34 |
+
"29": "lime",
|
35 |
+
"3": "bell_pepper",
|
36 |
+
"30": "marker",
|
37 |
+
"31": "mushroom",
|
38 |
+
"32": "notebook",
|
39 |
+
"33": "onion",
|
40 |
+
"34": "orange",
|
41 |
+
"35": "peach",
|
42 |
+
"36": "pear",
|
43 |
+
"37": "pitcher",
|
44 |
+
"38": "plate",
|
45 |
+
"39": "pliers",
|
46 |
+
"4": "binder",
|
47 |
+
"40": "potato",
|
48 |
+
"41": "rubber_eraser",
|
49 |
+
"42": "scissors",
|
50 |
+
"43": "shampoo",
|
51 |
+
"44": "soda_can",
|
52 |
+
"45": "sponge",
|
53 |
+
"46": "stapler",
|
54 |
+
"47": "tomato",
|
55 |
+
"48": "toothbrush",
|
56 |
+
"49": "toothpaste",
|
57 |
+
"5": "bowl",
|
58 |
+
"50": "water_bottle",
|
59 |
"6": "calculator",
|
60 |
+
"7": "camera",
|
61 |
+
"8": "cap",
|
62 |
+
"9": "cell_phone"
|
63 |
},
|
64 |
"image_size": 224,
|
65 |
"initializer_range": 0.02,
|
66 |
"intermediate_size": 3072,
|
67 |
"label2id": {
|
68 |
+
"apple": "0",
|
69 |
+
"ball": "1",
|
70 |
+
"banana": "2",
|
71 |
+
"bell_pepper": "3",
|
72 |
+
"binder": "4",
|
73 |
+
"bowl": "5",
|
74 |
"calculator": "6",
|
75 |
+
"camera": "7",
|
76 |
+
"cap": "8",
|
77 |
+
"cell_phone": "9",
|
78 |
"cereal_box": "10",
|
79 |
+
"coffee_mug": "11",
|
80 |
+
"comb": "12",
|
81 |
+
"dry_battery": "13",
|
82 |
+
"flashlight": "14",
|
83 |
+
"food_bag": "15",
|
84 |
+
"food_box": "16",
|
85 |
+
"food_can": "17",
|
86 |
+
"food_cup": "18",
|
87 |
+
"food_jar": "19",
|
88 |
+
"garlic": "20",
|
89 |
+
"glue_stick": "21",
|
90 |
+
"greens": "22",
|
91 |
+
"hand_towel": "23",
|
92 |
+
"instant_noodles": "24",
|
93 |
+
"keyboard": "25",
|
94 |
+
"kleenex": "26",
|
95 |
+
"lemon": "27",
|
96 |
+
"lightbulb": "28",
|
97 |
+
"lime": "29",
|
98 |
+
"marker": "30",
|
99 |
+
"mushroom": "31",
|
100 |
+
"notebook": "32",
|
101 |
+
"onion": "33",
|
102 |
+
"orange": "34",
|
103 |
+
"peach": "35",
|
104 |
+
"pear": "36",
|
105 |
+
"pitcher": "37",
|
106 |
+
"plate": "38",
|
107 |
+
"pliers": "39",
|
108 |
+
"potato": "40",
|
109 |
+
"rubber_eraser": "41",
|
110 |
+
"scissors": "42",
|
111 |
+
"shampoo": "43",
|
112 |
+
"soda_can": "44",
|
113 |
+
"sponge": "45",
|
114 |
+
"stapler": "46",
|
115 |
+
"tomato": "47",
|
116 |
+
"toothbrush": "48",
|
117 |
+
"toothpaste": "49",
|
118 |
+
"water_bottle": "50"
|
119 |
},
|
120 |
"layer_norm_eps": 1e-12,
|
121 |
"model_type": "vit",
|
logs/train/events.out.tfevents.1690226946.0dde144b55f7.2692.0.v2
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f54de307db6d5778d444ee8cfaddf8b1736a72f03d13637d19618dfb63681763
|
3 |
+
size 3307558
|
logs/validation/events.out.tfevents.1690233702.0dde144b55f7.2692.1.v2
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dbcbe17893a829375f5906e338102767afec6e4b1cc2167b916a6b4ca08a750a
|
3 |
+
size 568
|
tf_model.h5
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:164ba036d9807b05a74093fe6243e2b2e864ee137682ad78f180968ef0ec78c9
|
3 |
+
size 343620280
|