游雁 d16e9d8248 output_dir = None 2 lat temu
..
README.md 12a7adfdf3 update version 0.1.6 2 lat temu
finetune.py 12a7adfdf3 update version 0.1.6 2 lat temu
infer.py d16e9d8248 output_dir = None 2 lat temu

README.md

ModelScope Model

How to finetune and infer using a pretrained Paraformer-large Model

Finetune

  • Modify finetune training related parameters in finetune.py

    • output_dir: # result dir
    • data_dir: # the dataset dir needs to include files: train/wav.scp, train/text; validation/wav.scp, validation/text.
    • batch_bins: # batch size
    • max_epoch: # number of training epoch
    • lr: # learning rate
  • Then you can run the pipeline to finetune with:

    python finetune.py
    

Inference

Or you can use the finetuned model for inference directly.

  • Setting parameters in infer.py

    • audio_in: # support wav, url, bytes, and parsed audio format.
    • output_dir: # If the input format is wav.scp, it needs to be set.
  • Then you can run the pipeline to infer with:

    python infer.py