lucius-40 commited on
Commit
82dc6d9
·
verified ·
1 Parent(s): 063c9ba

End of training

Browse files
Files changed (2) hide show
  1. README.md +79 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ language:
4
+ - bn
5
+ license: mit
6
+ base_model: pyannote/segmentation-3.0
7
+ tags:
8
+ - audio
9
+ - speaker-diarization
10
+ - bengali
11
+ - pyannote
12
+ - speech
13
+ - generated_from_trainer
14
+ datasets:
15
+ - bengali-speaker-diarization
16
+ model-index:
17
+ - name: speaker-segmentation-bengali-optimized-conservative
18
+ results: []
19
+ ---
20
+
21
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
22
+ should probably proofread and complete it, then remove this comment. -->
23
+
24
+ # speaker-segmentation-bengali-optimized-conservative
25
+
26
+ This model is a fine-tuned version of [pyannote/segmentation-3.0](https://huggingface.co/pyannote/segmentation-3.0) on the bengali-speaker-diarization dataset.
27
+ It achieves the following results on the evaluation set:
28
+ - Loss: 0.4778
29
+ - Model Preparation Time: 0.0043
30
+ - Der: 0.1599
31
+ - False Alarm: 0.0403
32
+ - Missed Detection: 0.0162
33
+ - Confusion: 0.1034
34
+
35
+ ## Model description
36
+
37
+ More information needed
38
+
39
+ ## Intended uses & limitations
40
+
41
+ More information needed
42
+
43
+ ## Training and evaluation data
44
+
45
+ More information needed
46
+
47
+ ## Training procedure
48
+
49
+ ### Training hyperparameters
50
+
51
+ The following hyperparameters were used during training:
52
+ - learning_rate: 0.0005
53
+ - train_batch_size: 16
54
+ - eval_batch_size: 16
55
+ - seed: 42
56
+ - gradient_accumulation_steps: 2
57
+ - total_train_batch_size: 32
58
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
59
+ - lr_scheduler_type: cosine
60
+ - lr_scheduler_warmup_ratio: 0.1
61
+ - num_epochs: 10
62
+ - mixed_precision_training: Native AMP
63
+
64
+ ### Training results
65
+
66
+ | Training Loss | Epoch | Step | Validation Loss | Model Preparation Time | Der | False Alarm | Missed Detection | Confusion |
67
+ |:-------------:|:-----:|:----:|:---------------:|:----------------------:|:------:|:-----------:|:----------------:|:---------:|
68
+ | 0.4209 | 1.0 | 254 | 0.4600 | 0.0043 | 0.1574 | 0.0369 | 0.0185 | 0.1019 |
69
+ | 0.391 | 2.0 | 508 | 0.4628 | 0.0043 | 0.1586 | 0.0365 | 0.0215 | 0.1007 |
70
+ | 0.4302 | 3.0 | 762 | 0.4624 | 0.0043 | 0.1579 | 0.0388 | 0.0179 | 0.1012 |
71
+ | 0.3421 | 4.0 | 1016 | 0.4778 | 0.0043 | 0.1599 | 0.0403 | 0.0162 | 0.1034 |
72
+
73
+
74
+ ### Framework versions
75
+
76
+ - Transformers 4.48.3
77
+ - Pytorch 2.8.0+cu126
78
+ - Datasets 3.6.0
79
+ - Tokenizers 0.21.4
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:58389fe4ba7f5cc5a6905f9ed36c035bf020de91cd5ea43c2379a81687f97dbb
3
  size 5899124
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1aacc7dd1252dd5e21048f331c77bae36ba8d637236446b31be6b5638480acd8
3
  size 5899124