jolual2747 commited on
Commit
b20a0e7
1 Parent(s): 475700f

Model save

Browse files
README.md ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ model-index:
9
+ - name: vit-clothes-classification
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # vit-clothes-classification
17
+
18
+ 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.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 1.7642
21
+ - Accuracy: 0.6989
22
+
23
+ ## Model description
24
+
25
+ More information needed
26
+
27
+ ## Intended uses & limitations
28
+
29
+ More information needed
30
+
31
+ ## Training and evaluation data
32
+
33
+ More information needed
34
+
35
+ ## Training procedure
36
+
37
+ ### Training hyperparameters
38
+
39
+ The following hyperparameters were used during training:
40
+ - learning_rate: 0.0002
41
+ - train_batch_size: 8
42
+ - eval_batch_size: 8
43
+ - seed: 42
44
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
+ - lr_scheduler_type: linear
46
+ - num_epochs: 8
47
+
48
+ ### Training results
49
+
50
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
+ |:-------------:|:------:|:----:|:---------------:|:--------:|
52
+ | 1.0975 | 0.5714 | 500 | 1.2619 | 0.6111 |
53
+ | 0.8315 | 1.1429 | 1000 | 1.3133 | 0.6322 |
54
+ | 0.7266 | 1.7143 | 1500 | 1.2077 | 0.6356 |
55
+ | 0.5451 | 2.2857 | 2000 | 1.2895 | 0.6556 |
56
+ | 0.4287 | 2.8571 | 2500 | 1.2736 | 0.6644 |
57
+ | 0.2554 | 3.4286 | 3000 | 1.3801 | 0.6767 |
58
+ | 0.2265 | 4.0 | 3500 | 1.4924 | 0.6656 |
59
+ | 0.0738 | 4.5714 | 4000 | 1.6321 | 0.68 |
60
+ | 0.0761 | 5.1429 | 4500 | 1.6676 | 0.6767 |
61
+ | 0.0251 | 5.7143 | 5000 | 1.6911 | 0.7056 |
62
+ | 0.0147 | 6.2857 | 5500 | 1.7312 | 0.7 |
63
+ | 0.0051 | 6.8571 | 6000 | 1.7282 | 0.6922 |
64
+ | 0.0028 | 7.4286 | 6500 | 1.7679 | 0.6967 |
65
+ | 0.0017 | 8.0 | 7000 | 1.7642 | 0.6989 |
66
+
67
+
68
+ ### Framework versions
69
+
70
+ - Transformers 4.40.0
71
+ - Pytorch 2.2.1+cu121
72
+ - Datasets 2.19.0
73
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0f88df3a2c89bb4bb9c4df31e2137f95772d3e33a5c224125bb794cb7996dbbf
3
  size 343442380
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50ac16134e1bdc96ac2ce814fb96e22a6dbbaecb8095543fa7c69fdbf912ec1e
3
  size 343442380
runs/Apr24_04-18-24_ip-10-192-11-36/events.out.tfevents.1713932306.ip-10-192-11-36.16149.2 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9a34df59ab4d2e4c13730cb5096d82db4d7facdeb60d4a67062dd1ac8319b4b9
3
- size 15567
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:450dc7cc64e2a2d921119f8bcad70b6fe518521a4691d46f2172abb365b17fbe
3
+ size 15921