MinhLe999 commited on
Commit
e15c0ef
·
verified ·
1 Parent(s): 6c46347

End of training

Browse files
Files changed (4) hide show
  1. README.md +77 -0
  2. config.json +25 -0
  3. model.safetensors +3 -0
  4. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 3class_EfficientFormer30M_ForTesting
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
+ # 3class_EfficientFormer30M_ForTesting
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.0820
23
+ - Precision: 0.9831
24
+ - Recall: 0.9790
25
+ - Accuracy: 0.9844
26
+ - F1: 0.9809
27
+ - Roc Auc: 0.9981
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
+ - mixed_precision_training: Native AMP
54
+
55
+ ### Training results
56
+
57
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | Accuracy | F1 | Roc Auc |
58
+ |:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:--------:|:------:|:-------:|
59
+ | 0.1609 | 0.3436 | 200 | 0.0851 | 0.9673 | 0.9645 | 0.9718 | 0.9658 | 0.9964 |
60
+ | 0.0811 | 0.6873 | 400 | 0.0652 | 0.9767 | 0.9725 | 0.9790 | 0.9745 | 0.9981 |
61
+ | 0.0418 | 1.0309 | 600 | 0.1268 | 0.9602 | 0.9661 | 0.9690 | 0.9628 | 0.9948 |
62
+ | 0.0427 | 1.3746 | 800 | 0.0735 | 0.9756 | 0.9744 | 0.9793 | 0.9750 | 0.9973 |
63
+ | 0.0054 | 1.7182 | 1000 | 0.0614 | 0.9823 | 0.9779 | 0.9834 | 0.9800 | 0.9982 |
64
+ | 0.0293 | 2.0619 | 1200 | 0.0681 | 0.9812 | 0.9788 | 0.9834 | 0.9800 | 0.9983 |
65
+ | 0.0020 | 2.4055 | 1400 | 0.0676 | 0.9818 | 0.9787 | 0.9837 | 0.9802 | 0.9984 |
66
+ | 0.0332 | 2.7491 | 1600 | 0.0818 | 0.9812 | 0.9759 | 0.9823 | 0.9784 | 0.9982 |
67
+ | 0.0005 | 3.0928 | 1800 | 0.0755 | 0.9824 | 0.9793 | 0.9841 | 0.9808 | 0.9982 |
68
+ | 0.0017 | 3.4364 | 2000 | 0.0842 | 0.9819 | 0.9775 | 0.9832 | 0.9796 | 0.9979 |
69
+ | 0.0049 | 3.7801 | 2200 | 0.0820 | 0.9831 | 0.9790 | 0.9844 | 0.9809 | 0.9981 |
70
+
71
+
72
+ ### Framework versions
73
+
74
+ - Transformers 5.3.0
75
+ - Pytorch 2.10.0+cu128
76
+ - Datasets 4.0.0
77
+ - 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": "efficientformer_l3.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:393bd050c762da978db232c61c525527a22b12a773c87e9a328f9177ff945c4e
3
+ size 121893600
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc0e000f6c9c3a0df89fcef22aa4caf3c3f9ee2c6671e149493873b2296bbc86
3
+ size 5265