MinhLe999 commited on
Commit
769be3f
·
verified ·
1 Parent(s): 9411e58

End of training

Browse files
Files changed (4) hide show
  1. README.md +76 -0
  2. config.json +25 -0
  3. model.safetensors +3 -0
  4. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - precision
7
+ - recall
8
+ - accuracy
9
+ - f1
10
+ model-index:
11
+ - name: global_effformerv2_s2
12
+ results: []
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ # global_effformerv2_s2
19
+
20
+ This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
21
+ It achieves the following results on the evaluation set:
22
+ - Loss: 0.0634
23
+ - Precision: 0.9798
24
+ - Recall: 0.9761
25
+ - Accuracy: 0.9818
26
+ - F1: 0.9779
27
+ - Roc Auc: 0.9983
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 0.0001
47
+ - train_batch_size: 32
48
+ - eval_batch_size: 32
49
+ - seed: 42
50
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
51
+ - lr_scheduler_type: cosine
52
+ - num_epochs: 4
53
+
54
+ ### Training results
55
+
56
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | Accuracy | F1 | Roc Auc |
57
+ |:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:--------:|:------:|:-------:|
58
+ | 0.1269 | 0.3436 | 200 | 0.2372 | 0.9396 | 0.9473 | 0.9517 | 0.9427 | 0.9935 |
59
+ | 0.0940 | 0.6873 | 400 | 0.0662 | 0.9748 | 0.9710 | 0.9776 | 0.9728 | 0.9978 |
60
+ | 0.0869 | 1.0309 | 600 | 0.0805 | 0.9696 | 0.9675 | 0.9741 | 0.9685 | 0.9968 |
61
+ | 0.0401 | 1.3746 | 800 | 0.0649 | 0.9753 | 0.9742 | 0.9793 | 0.9748 | 0.9980 |
62
+ | 0.0455 | 1.7182 | 1000 | 0.0634 | 0.9798 | 0.9761 | 0.9818 | 0.9779 | 0.9983 |
63
+ | 0.0431 | 2.0619 | 1200 | 6.0823 | 0.9765 | 0.9771 | 0.9807 | 0.9768 | 0.9975 |
64
+ | 0.0259 | 2.4055 | 1400 | 0.0664 | 0.9802 | 0.9785 | 0.9830 | 0.9793 | 0.9983 |
65
+ | 0.0219 | 2.7491 | 1600 | 0.0703 | 0.9799 | 0.9760 | 0.9818 | 0.9779 | 0.9982 |
66
+ | 0.0128 | 3.0928 | 1800 | 0.0659 | 0.9797 | 0.9792 | 0.9830 | 0.9795 | 0.9982 |
67
+ | 0.0053 | 3.4364 | 2000 | 0.0667 | 0.9817 | 0.9792 | 0.9839 | 0.9804 | 0.9983 |
68
+ | 0.0045 | 3.7801 | 2200 | 0.0697 | 0.9809 | 0.9773 | 0.9828 | 0.9790 | 0.9982 |
69
+
70
+
71
+ ### Framework versions
72
+
73
+ - Transformers 5.3.0
74
+ - Pytorch 2.10.0+cu128
75
+ - Datasets 4.7.0
76
+ - Tokenizers 0.22.2
config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "UniversalVisionModel"
4
+ ],
5
+ "auto_map": {
6
+ "AutoConfig": "modeling_universal_vision.UniversalVisionConfig",
7
+ "AutoModelForImageClassification": "modeling_universal_vision.UniversalVisionModel"
8
+ },
9
+ "backbone_name": "efficientformerv2_s2.snap_dist_in1k",
10
+ "dtype": "float32",
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2"
15
+ },
16
+ "label2id": {
17
+ "LABEL_0": 0,
18
+ "LABEL_1": 1,
19
+ "LABEL_2": 2
20
+ },
21
+ "model_type": "universal_vision",
22
+ "pretrained": true,
23
+ "transformers_version": "5.3.0",
24
+ "use_cache": false
25
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de5d2f19cd7b4bf1d2a4acd0191d97469497e781b1e7db2a1f45ad7a71285ea3
3
+ size 49145824
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14a00fdbe69acbdb95eb7d39d2e7962376b837970643608fdef6b26c099600d8
3
+ size 5265