End of training

README.md

@@ -22,12 +22,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [pyannote/speaker-diarization-3.1](https://huggingface.co/pyannote/speaker-diarization-3.1) on the speaker-segmentation dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3831
-- Model Preparation Time: 0.0038
-- Der: 0.1279
-- False Alarm: 0.0209
-- Missed Detection: 0.0105
-- Confusion: 0.0965
+- Loss: 0.3735
+- Model Preparation Time: 0.0099
+- Der: 0.1228
+- False Alarm: 0.0205
+- Missed Detection: 0.0096
+- Confusion: 0.0927
 
 ## Model description
 
@@ -52,17 +52,22 @@ The following hyperparameters were used during training:
 - seed: 43
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: cosine
-- num_epochs: 5
+- num_epochs: 10
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Model Preparation Time | Der    | False Alarm | Missed Detection | Confusion |
 |:-------------:|:-----:|:----:|:---------------:|:----------------------:|:------:|:-----------:|:----------------:|:---------:|
-| 0.4394        | 1.0   | 570  | 0.4349          | 0.0038                 | 0.1462 | 0.0209      | 0.0115           | 0.1138    |
-| 0.3989        | 2.0   | 1140 | 0.4134          | 0.0038                 | 0.1385 | 0.0211      | 0.0107           | 0.1067    |
-| 0.3797        | 3.0   | 1710 | 0.3871          | 0.0038                 | 0.1302 | 0.0211      | 0.0104           | 0.0986    |
-| 0.3518        | 4.0   | 2280 | 0.3817          | 0.0038                 | 0.1263 | 0.0209      | 0.0106           | 0.0948    |
-| 0.3269        | 5.0   | 2850 | 0.3831          | 0.0038                 | 0.1279 | 0.0209      | 0.0105           | 0.0965    |
+| 0.4492        | 1.0   | 570  | 0.4525          | 0.0099                 | 0.1515 | 0.0219      | 0.0094           | 0.1202    |
+| 0.3875        | 2.0   | 1140 | 0.4229          | 0.0099                 | 0.1403 | 0.0212      | 0.0098           | 0.1093    |
+| 0.4014        | 3.0   | 1710 | 0.3889          | 0.0099                 | 0.1324 | 0.0207      | 0.0102           | 0.1016    |
+| 0.3656        | 4.0   | 2280 | 0.3899          | 0.0099                 | 0.1305 | 0.0212      | 0.0094           | 0.1000    |
+| 0.3232        | 5.0   | 2850 | 0.3804          | 0.0099                 | 0.1283 | 0.0207      | 0.0096           | 0.0981    |
+| 0.319         | 6.0   | 3420 | 0.3796          | 0.0099                 | 0.1256 | 0.0204      | 0.0097           | 0.0956    |
+| 0.3018        | 7.0   | 3990 | 0.3750          | 0.0099                 | 0.1247 | 0.0202      | 0.0101           | 0.0945    |
+| 0.2928        | 8.0   | 4560 | 0.3700          | 0.0099                 | 0.1220 | 0.0203      | 0.0097           | 0.0920    |
+| 0.3321        | 9.0   | 5130 | 0.3753          | 0.0099                 | 0.1238 | 0.0204      | 0.0098           | 0.0936    |
+| 0.2738        | 10.0  | 5700 | 0.3735          | 0.0099                 | 0.1228 | 0.0205      | 0.0096           | 0.0927    |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f19579ccd85d075e2fcb581a858fd1716d8833f88c6142947ebc2c394e2dcb89
+oid sha256:f64b44eea1ef3651b5b3a7185f378dc069dfc5a2eaf391c1ccab64e6fe37c5a6
 size 5899124

runs/Apr10_02-22-44_2e7618420631/events.out.tfevents.1744251792.2e7618420631.2507.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ca4148427ece742ba13741607934988ff707b3e9392d3c4f52b03cf0c9c198c4
-size 53401
+oid sha256:4ff4e43b5646896cc032b81378b1f870c23be30e033ee281010a8cc62361dc4d
+size 59160