ideepankarsharma2003 commited on
Commit
5adad7a
·
verified ·
1 Parent(s): 65cbba0

End of training

Browse files
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google/vit-base-patch16-224-in21k
5
+ tags:
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: clothing-multilabel
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # clothing-multilabel
16
+
17
+ This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the None dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 0.3360
20
+ - Exact Match Accuracy: 0.7919
21
+
22
+ ## Model description
23
+
24
+ More information needed
25
+
26
+ ## Intended uses & limitations
27
+
28
+ More information needed
29
+
30
+ ## Training and evaluation data
31
+
32
+ More information needed
33
+
34
+ ## Training procedure
35
+
36
+ ### Training hyperparameters
37
+
38
+ The following hyperparameters were used during training:
39
+ - learning_rate: 5e-05
40
+ - train_batch_size: 32
41
+ - eval_batch_size: 32
42
+ - seed: 42
43
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
44
+ - lr_scheduler_type: linear
45
+ - lr_scheduler_warmup_ratio: 0.1
46
+ - num_epochs: 3
47
+
48
+ ### Training results
49
+
50
+ | Training Loss | Epoch | Step | Validation Loss | Exact Match Accuracy |
51
+ |:-------------:|:-----:|:----:|:---------------:|:--------------------:|
52
+ | 0.2955 | 1.0 | 1864 | 0.2995 | 0.7884 |
53
+ | 0.228 | 2.0 | 3728 | 0.2937 | 0.7958 |
54
+ | 0.117 | 3.0 | 5592 | 0.3360 | 0.7919 |
55
+
56
+
57
+ ### Framework versions
58
+
59
+ - Transformers 4.52.3
60
+ - Pytorch 2.7.0+cu126
61
+ - Datasets 3.6.0
62
+ - Tokenizers 0.21.1
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViTForImageClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "encoder_stride": 16,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.0,
9
+ "hidden_size": 768,
10
+ "id2label": {
11
+ "0": "longpants",
12
+ "1": "longsleeves"
13
+ },
14
+ "image_size": 224,
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 3072,
17
+ "label2id": {
18
+ "longpants": 0,
19
+ "longsleeves": 1
20
+ },
21
+ "layer_norm_eps": 1e-12,
22
+ "model_type": "vit",
23
+ "num_attention_heads": 12,
24
+ "num_channels": 3,
25
+ "num_hidden_layers": 12,
26
+ "patch_size": 16,
27
+ "pooler_act": "tanh",
28
+ "pooler_output_size": 768,
29
+ "problem_type": "multi_label_classification",
30
+ "qkv_bias": true,
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.52.3"
33
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db028f829fd864d71d8b6875864fd7ec03e4dcc64075e6d6333c3cc4a2268144
3
+ size 343223968
preprocessor_config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": null,
3
+ "data_format": "channels_first",
4
+ "default_to_square": true,
5
+ "device": null,
6
+ "do_center_crop": null,
7
+ "do_convert_rgb": null,
8
+ "do_normalize": true,
9
+ "do_rescale": true,
10
+ "do_resize": true,
11
+ "image_mean": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "image_processor_type": "ViTImageProcessorFast",
17
+ "image_std": [
18
+ 0.5,
19
+ 0.5,
20
+ 0.5
21
+ ],
22
+ "input_data_format": null,
23
+ "resample": 2,
24
+ "rescale_factor": 0.00392156862745098,
25
+ "return_tensors": null,
26
+ "size": {
27
+ "height": 224,
28
+ "width": 224
29
+ }
30
+ }
runs/Jun02_14-52-47_firefly/events.out.tfevents.1748856168.firefly.121671.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e2b7976f4e815b3eccdc05cb30d9259c23aad74e5a6a3039ec3ec3efecff4a8
3
+ size 29222
runs/Jun02_15-05-14_firefly/events.out.tfevents.1748856915.firefly.133108.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9182b2dc2dbe0fe73d52ac162dcd3fa03657bbd0a66d48710d0f8e90b26136ca
3
+ size 124476
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92120a7d786876223370b9222b80c676dc63a12cff9f7159fa16cd40bafce365
3
+ size 5841