skutaada commited on
Commit
17b521b
1 Parent(s): 2cd2645

End of training

Browse files
Files changed (2) hide show
  1. README.md +78 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - generated_from_trainer
4
+ metrics:
5
+ - accuracy
6
+ model-index:
7
+ - name: MobileNetV2-KD-VGGFace
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # MobileNetV2-KD-VGGFace
15
+
16
+ This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 0.4919
19
+ - Accuracy: 0.7836
20
+
21
+ ## Model description
22
+
23
+ More information needed
24
+
25
+ ## Intended uses & limitations
26
+
27
+ More information needed
28
+
29
+ ## Training and evaluation data
30
+
31
+ More information needed
32
+
33
+ ## Training procedure
34
+
35
+ ### Training hyperparameters
36
+
37
+ The following hyperparameters were used during training:
38
+ - learning_rate: 5e-05
39
+ - train_batch_size: 24
40
+ - eval_batch_size: 24
41
+ - seed: 42
42
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
+ - lr_scheduler_type: linear
44
+ - num_epochs: 20
45
+ - mixed_precision_training: Native AMP
46
+
47
+ ### Training results
48
+
49
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|
51
+ | 2.7506 | 1.0 | 1667 | 2.2449 | 0.0726 |
52
+ | 2.105 | 2.0 | 3334 | 1.6904 | 0.2493 |
53
+ | 1.6544 | 3.0 | 5001 | 1.3206 | 0.4043 |
54
+ | 1.3357 | 4.0 | 6668 | 1.0675 | 0.5078 |
55
+ | 1.1104 | 5.0 | 8335 | 0.9302 | 0.5582 |
56
+ | 0.9287 | 6.0 | 10002 | 0.8738 | 0.5972 |
57
+ | 0.7899 | 7.0 | 11669 | 0.7972 | 0.6388 |
58
+ | 0.6738 | 8.0 | 13336 | 0.7074 | 0.6822 |
59
+ | 0.5803 | 9.0 | 15003 | 0.6630 | 0.7009 |
60
+ | 0.5038 | 10.0 | 16670 | 0.5855 | 0.735 |
61
+ | 0.4366 | 11.0 | 18337 | 0.5761 | 0.7415 |
62
+ | 0.3762 | 12.0 | 20004 | 0.5642 | 0.7496 |
63
+ | 0.3321 | 13.0 | 21671 | 0.5373 | 0.7652 |
64
+ | 0.2916 | 14.0 | 23338 | 0.5314 | 0.7625 |
65
+ | 0.2615 | 15.0 | 25005 | 0.6206 | 0.7281 |
66
+ | 0.2357 | 16.0 | 26672 | 0.5437 | 0.763 |
67
+ | 0.2153 | 17.0 | 28339 | 0.5335 | 0.763 |
68
+ | 0.1986 | 18.0 | 30006 | 0.4892 | 0.7869 |
69
+ | 0.1866 | 19.0 | 31673 | 0.5368 | 0.7645 |
70
+ | 0.1765 | 20.0 | 33340 | 0.4919 | 0.7836 |
71
+
72
+
73
+ ### Framework versions
74
+
75
+ - Transformers 4.42.4
76
+ - Pytorch 2.3.1+rocm6.0
77
+ - Datasets 2.20.0
78
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b246fc5a8a33f28a8c9e48ce5236b34836eb5366e749a67495b4ff606b199b8d
3
  size 55857748
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4bbfb955818e368de5e29ca4eb617e7fc9a216a738cacf031a8778ad398ef00b
3
  size 55857748