End of training

README.md

@@ -5,24 +5,9 @@ tags:
 - generated_from_trainer
 datasets:
 - marsyas/gtzan
-metrics:
-- accuracy
 model-index:
 - name: distilhubert-finetuned-gtzan
-  results:
-  - task:
-      name: Audio Classification
-      type: audio-classification
-    dataset:
-      name: GTZAN
-      type: marsyas/gtzan
-      config: all
-      split: train
-      args: all
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.83
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +17,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [arshsin/distilhubert-finetuned-gtzan](https://huggingface.co/arshsin/distilhubert-finetuned-gtzan) on the GTZAN dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.1522
-- Accuracy: 0.83
+- eval_loss: 1.3357
+- eval_accuracy: 0.84
+- eval_runtime: 72.1033
+- eval_samples_per_second: 1.387
+- eval_steps_per_second: 0.236
+- epoch: 3.97
+- step: 85
 
 ## Model description
 
@@ -52,29 +42,18 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1e-05
-- train_batch_size: 8
-- eval_batch_size: 8
+- learning_rate: 4e-05
+- train_batch_size: 6
+- eval_batch_size: 6
 - seed: 42
-- gradient_accumulation_steps: 4
-- total_train_batch_size: 32
+- gradient_accumulation_steps: 7
+- total_train_batch_size: 42
 - 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
+- num_epochs: 10
 - mixed_precision_training: Native AMP
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.0017        | 0.99  | 28   | 1.0910          | 0.82     |
-| 0.0143        | 1.98  | 56   | 1.1379          | 0.83     |
-| 0.01          | 2.97  | 84   | 1.1370          | 0.82     |
-| 0.0377        | 4.0   | 113  | 1.1428          | 0.82     |
-| 0.0318        | 4.96  | 140  | 1.1522          | 0.83     |
-
-
 ### Framework versions
 
 - Transformers 4.35.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d29ea7d54fd5609806bbc21054a0eaf918036aa5c0ccd4a443ffde67e9d14cd7
+oid sha256:a52c2baf3403cec14110a834770fcefd8cc3111bdd09d170f675e6056e9d30ef
 size 94771728

runs/Feb12_18-05-36_720eab1d60a4/events.out.tfevents.1707761147.720eab1d60a4.34.10

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e9c3709c01c3cd7a990b29845b4c46dbb8e1ff47c7dc6ba92b141c74aa85c2b2
-size 9357
+oid sha256:22ebdf845ddc1fdfe76b5d20c37c717515c4ffccde88bf8b0d1f2caf9f61a41c
+size 9511