aalonso-developer commited on
Commit
7b11329
1 Parent(s): 86e21a9

Training in progress epoch 0

Browse files
README.md ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_keras_callback
5
+ model-index:
6
+ - name: aalonso-developer/vit-base-patch16-224-in21k-clothing-classifier
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information Keras had access to. You should
11
+ probably proofread and complete it, then remove this comment. -->
12
+
13
+ # aalonso-developer/vit-base-patch16-224-in21k-clothing-classifier
14
+
15
+ 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.
16
+ It achieves the following results on the evaluation set:
17
+ - Train Loss: 1.3672
18
+ - Train Accuracy: 0.6720
19
+ - Train Top-3-accuracy: 0.8895
20
+ - Validation Loss: 0.8886
21
+ - Validation Accuracy: 0.7613
22
+ - Validation Top-3-accuracy: 0.9415
23
+ - Epoch: 0
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 3665, '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}
43
+ - training_precision: mixed_float16
44
+
45
+ ### Training results
46
+
47
+ | Train Loss | Train Accuracy | Train Top-3-accuracy | Validation Loss | Validation Accuracy | Validation Top-3-accuracy | Epoch |
48
+ |:----------:|:--------------:|:--------------------:|:---------------:|:-------------------:|:-------------------------:|:-----:|
49
+ | 1.3672 | 0.6720 | 0.8895 | 0.8886 | 0.7613 | 0.9415 | 0 |
50
+
51
+
52
+ ### Framework versions
53
+
54
+ - Transformers 4.29.1
55
+ - TensorFlow 2.11.0
56
+ - Datasets 2.12.0
57
+ - Tokenizers 0.13.3
config.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "Skirt",
13
+ "1": "Jacket",
14
+ "10": "Cardigan",
15
+ "11": "Jumpsuit",
16
+ "12": "Sweater",
17
+ "13": "Blazer",
18
+ "2": "Dress",
19
+ "3": "Tee",
20
+ "4": "Sweatpants",
21
+ "5": "Tank",
22
+ "6": "Romper",
23
+ "7": "Top",
24
+ "8": "Shorts",
25
+ "9": "Jeans"
26
+ },
27
+ "image_size": 224,
28
+ "initializer_range": 0.02,
29
+ "intermediate_size": 3072,
30
+ "label2id": {
31
+ "Blazer": "13",
32
+ "Cardigan": "10",
33
+ "Dress": "2",
34
+ "Jacket": "1",
35
+ "Jeans": "9",
36
+ "Jumpsuit": "11",
37
+ "Romper": "6",
38
+ "Shorts": "8",
39
+ "Skirt": "0",
40
+ "Sweater": "12",
41
+ "Sweatpants": "4",
42
+ "Tank": "5",
43
+ "Tee": "3",
44
+ "Top": "7"
45
+ },
46
+ "layer_norm_eps": 1e-12,
47
+ "model_type": "vit",
48
+ "num_attention_heads": 12,
49
+ "num_channels": 3,
50
+ "num_hidden_layers": 12,
51
+ "patch_size": 16,
52
+ "qkv_bias": true,
53
+ "transformers_version": "4.29.1"
54
+ }
logs/train/events.out.tfevents.1684085132.adrian-development.12522.0.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70513f90c1055fd572d524253a35ca4630dc43da606e2a4d0563be15d581e53d
3
+ size 3332192
logs/validation/events.out.tfevents.1684085494.adrian-development.12522.1.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ef0c4ce3ed40ff83cc3d18d1f7d7bb34cbe758e4f53c3393c0b31d0af4fbdd5
3
+ size 530
tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4942328dd17ddadb170349956b2470158148a102eadce63b1325c7bf7f5e94e
3
+ size 343523000