Skip to content

littlebeanhp/ZAC2022-LightningConformer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ZAC2022-LightningConformer

Team Lightning's solution for Zalo AI Challenge 2022

Dataset

Link:

https://drive.google.com/file/d/1vUSxdORDxk-ePUt-bUVDahpoXiqKchMx/

Note that users should create a metadata file in .json format with audio_filepath, transcript and note metadata fields.

Prepare environment

pip install -r requirements.txt

Train the model

We pre-trained the model with self-supervised approach on Vin BigData ASR

python train.py --config-name pretrain.yaml

After that, this command is to fine-tune the model with ZAC2022 train dataset (Optional)

python train.py --config-name finetune.yaml

Inference

Model Inference can be simply performed with one line

python inference.py

About

Team Lightning's solution for Zalo AI Challenge 2022

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages