File size: 830 Bytes
99890d7
e9bc91d
99890d7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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: ..