Model save

README.md

@@ -20,18 +20,18 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [Bisher/wav2vec2_ASV_deepfake_audio_detection](https://huggingface.co/Bisher/wav2vec2_ASV_deepfake_audio_detection) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5551
-- Accuracy: 0.89
-- Precision: 0.8815
-- Recall: 0.89
-- F1: 0.8667
-- Tp: 4
-- Tn: 85
-- Fn: 10
-- Fp: 1
-- Eer: 0.1860
-- Min Tdcf: 0.0429
-- Auc Roc: 0.9203
+- Loss: 0.3702
+- Accuracy: 0.9173
+- Precision: 0.9193
+- Recall: 0.9173
+- F1: 0.8917
+- Tp: 384
+- Tn: 17889
+- Fn: 1623
+- Fp: 24
+- Eer: 0.0762
+- Min Tdcf: 0.0323
+- Auc Roc: 0.9561
 
 ## Model description
 
@@ -50,23 +50,26 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
-- train_batch_size: 2
-- eval_batch_size: 2
+- learning_rate: 3e-05
+- train_batch_size: 152
+- eval_batch_size: 152
 - seed: 42
 - gradient_accumulation_steps: 4
-- total_train_batch_size: 8
+- total_train_batch_size: 608
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 1
+- num_epochs: 5
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     | Tp | Tn | Fn | Fp | Eer    | Min Tdcf | Auc Roc |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|:--:|:--:|:--:|:--:|:------:|:--------:|:-------:|
-| 0.582         | 0.8   | 10   | 0.5551          | 0.89     | 0.8815    | 0.89   | 0.8667 | 4  | 85 | 10 | 1  | 0.1860 | 0.0429   | 0.9203  |
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy | Precision | Recall | F1     | Tp  | Tn    | Fn   | Fp | Eer    | Min Tdcf | Auc Roc |
+|:-------------:|:------:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|:---:|:-----:|:----:|:--:|:------:|:--------:|:-------:|
+| 0.0237        | 0.0490 | 10   | 0.3313          | 0.9162   | 0.9174    | 0.9162 | 0.8897 | 364 | 17886 | 1643 | 27 | 0.0829 | 0.0335   | 0.9725  |
+| 0.0219        | 0.0979 | 20   | 0.3291          | 0.9147   | 0.9168    | 0.9147 | 0.8865 | 328 | 17892 | 1679 | 21 | 0.0809 | 0.0328   | 0.9736  |
+| 0.0202        | 0.1469 | 30   | 0.3233          | 0.9173   | 0.9189    | 0.9173 | 0.8918 | 386 | 17887 | 1621 | 26 | 0.0765 | 0.0323   | 0.9738  |
+| 0.0205        | 0.1958 | 40   | 0.3702          | 0.9173   | 0.9193    | 0.9173 | 0.8917 | 384 | 17889 | 1623 | 24 | 0.0762 | 0.0323   | 0.9561  |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1c12a46727745b3f22517ad35813106f5a40dc85fac08989ebd186a62c7b8d44
+oid sha256:3849fd57956b08273d6d9aa3ba1eb20404f6f779f0b2f14d155a64475fe5b842
 size 378302360