NbAiLabArchive/scream_sextusdecimus_virtual_tsfix_small

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

screamsextusdecimusvirtualtsfixsmall

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

• —step: 19999
• —eval_loss: 0.2913
• —train_loss: 0.6610
• —eval_wer: 8.7151
• —eval_cer: 3.8962
• —evalexactwer: 8.7151
• —evalexactcer: 3.8962

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: 5e-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: 11920
• —num_beams: 5
• —dropout: True
• —bpedropoutprobability: 0.1
• —activationdropoutprobability: 0.1

Training results

Framework versions

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