whisper-accent-medium.en-accent-head-only

This model is a fine-tuned version of openai/whisper-medium.en on the westbrook/English_Accent_DataSet dataset. It achieves the following results on the evaluation set:

• Loss: 0.1134
• Accent Accuracy: 0.9615

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 0.001
• train_batch_size: 4
• eval_batch_size: 4
• seed: 42
• distributed_type: multi-GPU
• num_devices: 2
• gradient_accumulation_steps: 4
• total_train_batch_size: 32
• total_eval_batch_size: 8
• 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: linear
• lr_scheduler_warmup_steps: 0.05
• num_epochs: 2.0

Training results

Training Loss	Epoch	Step	Validation Loss	Accent Accuracy
No log	0	0	3.1380	0.0058
2.1517	0.1270	200	2.0984	0.1269
1.4907	0.2541	400	1.1143	0.5788
0.8626	0.3811	600	0.5538	0.8115
0.6133	0.5082	800	0.3324	0.8865
0.4397	0.6352	1000	0.3983	0.8615
0.4342	0.7623	1200	0.3863	0.8865
0.3086	0.8893	1400	0.2686	0.9231
0.2464	1.0159	1600	0.1967	0.9135
0.2900	1.1429	1800	0.1944	0.9365
0.2410	1.2700	2000	0.1634	0.9423
0.2188	1.3970	2200	0.1401	0.9558
0.1722	1.5241	2400	0.1484	0.9596
0.1014	1.6511	2600	0.1222	0.9615
0.1926	1.7781	2800	0.1099	0.9577
0.0768	1.9052	3000	0.1013	0.9577

Framework versions

• Transformers 5.2.0
• Pytorch 2.10.0+cu128
• Datasets 4.5.0
• Tokenizers 0.22.2