Model save

README.md

@@ -2,7 +2,6 @@
 license: apache-2.0
 base_model: facebook/wav2vec2-base
 tags:
-- audio-classification
 - generated_from_trainer
 metrics:
 - accuracy
@@ -16,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # wav2vec2-base-ft-fake-detection
 
-This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the alexandreacff/kaggle-fake-detection dataset.
+This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2780
-- Accuracy: 0.9907
+- Loss: 0.8924
+- Accuracy: 0.6178
 
 ## Model description
 
@@ -47,23 +46,28 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 5.0
+- num_epochs: 10.0
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| 0.4897        | 0.9851 | 33   | 1.3925          | 0.0      |
-| 0.3905        | 2.0    | 67   | 0.6338          | 0.7953   |
-| 0.3139        | 2.9851 | 100  | 0.4037          | 0.9710   |
-| 0.2777        | 4.0    | 134  | 0.3067          | 0.9888   |
-| 0.2455        | 4.9254 | 165  | 0.2780          | 0.9907   |
+| 0.6253        | 0.9851 | 33   | 0.6261          | 0.6523   |
+| 0.4394        | 2.0    | 67   | 0.7140          | 0.5645   |
+| 0.3685        | 2.9851 | 100  | 0.7181          | 0.5850   |
+| 0.317         | 4.0    | 134  | 0.7291          | 0.6150   |
+| 0.3027        | 4.9851 | 167  | 0.7457          | 0.6159   |
+| 0.2672        | 6.0    | 201  | 0.7805          | 0.6243   |
+| 0.2711        | 6.9851 | 234  | 0.8113          | 0.6215   |
+| 0.2086        | 8.0    | 268  | 0.9130          | 0.5963   |
+| 0.2077        | 8.9851 | 301  | 0.9042          | 0.6168   |
+| 0.223         | 9.8507 | 330  | 0.8924          | 0.6178   |
 
 
 ### Framework versions
 
 - Transformers 4.41.0.dev0
-- Pytorch 2.2.1+cu121
+- Pytorch 2.1.0a0+32f93b1
 - Datasets 2.19.1
 - Tokenizers 0.19.1

model.safetensors

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