NbAiLabArchive/scream_tertius_dropout_replicate_test7b

<!-- This model card has been generated automatically according to the information Keras had access to. You should probably proofread and complete it, then remove this comment. -->

screamtertiusdropoutreplicatetest7b

This model is a fine-tuned version of openai/whisper-small on the NbAiLab/NCCspeechall_v5 dataset. It achieves the following results on the evaluation set:

• —step: 19999
• —eval_loss: 0.6607
• —train_loss: 0.3094
• —eval_wer: 10.8709
• —eval_cer: 5.1449

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• —learning_rate: 2e-05
• —lrschedulertype: linear
• —perdevicetrainbatchsize: 32
• —totaltrainbatchsizeper_node: 128
• —totaltrainbatch_size: 1024
• —totaloptimizationsteps: 20,000
• —startingoptimizationstep: None
• —finishingoptimizationstep: 20,000
• —numtraindataset_workers: 32
• —num_hosts: 8
• —totalnumtraining_examples: 20,480,000
• —stepsperepoch: 1314
• —num_beams: 5
• —dropout: True
• —dropout_probability: 0.1

Training results

Framework versions

• —Transformers 4.30.0.dev0
• —Datasets 2.12.0
• —Tokenizers 0.13.3