NbAiLabArchive/scream_tertius_dropout_replicate_test7a

<!-- 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. -->

screamtertiusdropoutreplicatetest7a

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: 1.1799
• —train_loss: 0.3145
• —eval_wer: 11.6626
• —eval_cer: 5.8907

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.29.0.dev0
• —Datasets 2.12.0
• —Tokenizers 0.13.3