latest-demo / conf /gradio-asr.yaml
AnasPwnapple's picture
Use CPU
e9bc91d
raw
history blame
No virus
830 Bytes
sample_rate: &sample_rate 16000
device: &device "cpu"
gradio:
source: "upload" # 'upload' or 'microphone'
title: "CitriNet(Lg) Un/Diacritized Comparison"
models:
- _target_: "models_evaluation.models.citrinet.CitrinetTorch"
cfg:
model_target: "nemo.collections.asr.models.EncDecCTCModelBPE"
name: "CnLg_SpeUni1024_DI_EATL600"
sample_rate: *sample_rate
device: *device
model_path: "wandb://tarteel/nemo-experiments/CnLg_SpeUni1024_DI_EATL600:v1"
- _target_: "models_evaluation.models.citrinet.CitrinetTorch"
cfg:
model_target: "nemo.collections.asr.models.EncDecCTCModelBPE"
name: "CnLg-SpeUni256-EATL1300"
sample_rate: *sample_rate
device: *device
model_path: "wandb://tarteel/nemo-experiments/CnLg-SpeUni256-EATL1300:v1"
hydra:
run:
dir: ..