Skip to content

i-vainn/speech2text

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

47 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ASR project barebones

Installation guide

  1. Build and start docker container.
   docker build -t speech2text .
   docker run \
   --gpus '"device=0"' \
   -it --rm speech2text
  1. Download best model checkpoint from https://disk.yandex.ru/d/nrjR_EMvW5wfSA or with script.
wget  -O model_best.pth https://file.io/X8k7zGWAg8y5
  1. Download LM checkpoint.
wget https://www.openslr.org/resources/11/4-gram.arpa.gz --no-check-certificate
  1. Inference model!
   python test.py -c hw_asr/configs/test_config.json -t test_data -b 5 -r model_best.pth -o output_other.json

or

  python test.py -c hw_asr/configs/test_config.json -b 5 -r model_best.pth -o output_other.json

which runs eval on test-other by default.

  1. Train?
   python train.py -c hw_asr/config.json

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.6%
  • Dockerfile 0.4%