sagir567 commited on
Commit
c89946d
1 Parent(s): a321e62

End of training

Browse files
Files changed (4) hide show
  1. README.md +78 -0
  2. config.json +1 -1
  3. model.safetensors +2 -2
  4. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: facebook/wav2vec2-base
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ - precision
10
+ - recall
11
+ - f1
12
+ model-index:
13
+ - name: wav2vec2transformerEMR3
14
+ results: []
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+ # wav2vec2transformerEMR3
21
+
22
+ This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on an unknown dataset.
23
+ It achieves the following results on the evaluation set:
24
+ - Loss: 0.6589
25
+ - Accuracy: 0.7916
26
+ - Precision: 0.7918
27
+ - Recall: 0.7916
28
+ - F1: 0.7896
29
+
30
+ ## Model description
31
+
32
+ More information needed
33
+
34
+ ## Intended uses & limitations
35
+
36
+ More information needed
37
+
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
+
42
+ ## Training procedure
43
+
44
+ ### Training hyperparameters
45
+
46
+ The following hyperparameters were used during training:
47
+ - learning_rate: 1e-05
48
+ - train_batch_size: 16
49
+ - eval_batch_size: 16
50
+ - seed: 42
51
+ - gradient_accumulation_steps: 2
52
+ - total_train_batch_size: 32
53
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
54
+ - lr_scheduler_type: linear
55
+ - lr_scheduler_warmup_steps: 500
56
+ - num_epochs: 15
57
+ - mixed_precision_training: Native AMP
58
+
59
+ ### Training results
60
+
61
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
62
+ |:-------------:|:-------:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
63
+ | 1.4706 | 1.6420 | 500 | 1.4207 | 0.5409 | 0.5559 | 0.5409 | 0.5180 |
64
+ | 0.953 | 3.2841 | 1000 | 0.9292 | 0.7275 | 0.7438 | 0.7275 | 0.7233 |
65
+ | 0.7575 | 4.9261 | 1500 | 0.7618 | 0.7616 | 0.7686 | 0.7616 | 0.7610 |
66
+ | 0.6084 | 6.5681 | 2000 | 0.7485 | 0.7559 | 0.7658 | 0.7559 | 0.7524 |
67
+ | 0.5221 | 8.2102 | 2500 | 0.6990 | 0.7711 | 0.7767 | 0.7711 | 0.7691 |
68
+ | 0.431 | 9.8522 | 3000 | 0.6967 | 0.7752 | 0.7796 | 0.7752 | 0.7719 |
69
+ | 0.3814 | 11.4943 | 3500 | 0.6523 | 0.7867 | 0.7875 | 0.7867 | 0.7856 |
70
+ | 0.3461 | 13.1363 | 4000 | 0.6589 | 0.7916 | 0.7918 | 0.7916 | 0.7896 |
71
+ | 0.3405 | 14.7783 | 4500 | 0.6703 | 0.7867 | 0.7878 | 0.7867 | 0.7847 |
72
+
73
+
74
+ ### Framework versions
75
+
76
+ - Transformers 4.46.2
77
+ - Pytorch 2.5.1+cu121
78
+ - Tokenizers 0.20.3
config.json CHANGED
@@ -7,7 +7,7 @@
7
  "add_adapter": false,
8
  "apply_spec_augment": true,
9
  "architectures": [
10
- "Wav2Vec2Model"
11
  ],
12
  "attention_dropout": 0.1,
13
  "bos_token_id": 1,
 
7
  "add_adapter": false,
8
  "apply_spec_augment": true,
9
  "architectures": [
10
+ "Wav2Vec2ForSequenceClassification"
11
  ],
12
  "attention_dropout": 0.1,
13
  "bos_token_id": 1,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fa7eeb43c6df4d1a3c64268f57881e75d36bbc9d3aa2dd57bf522600c9f3525a
3
- size 377510632
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a97f46672eb03a8603484715488a46395e67646d47af4ede6c81ade215bb7d62
3
+ size 378307508
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88f738b0de0e2ec08cde2bb932bdc6de703bd7b78dfdefd9e9aefb0c7fe27ac0
3
+ size 5240