experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove hardcoded constants from evaluate_tflite.py

Browse Source
This commit is contained in:
Reuben Morais 2020-02-12 13:08:34 +01:00
parent 3637f88c06
commit c2f336b3e3
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
evaluate_tflite.py
View File

@ -30,16 +30,9 @@ This module should be self-contained:
Then run with a TF Lite model, a scorer and a CSV test file Then run with a TF Lite model, a scorer and a CSV test file
''' '''
BEAM_WIDTH = 500
LM_ALPHA = 0.75
LM_BETA = 1.85
def tflite_worker(model, scorer, queue_in, queue_out, gpu_mask): def tflite_worker(model, scorer, queue_in, queue_out, gpu_mask):
os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_mask) os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_mask)
ds = Model(model) ds = Model(model)
ds.setBeamWidth(BEAM_WIDTH)
ds.enableExternalScorer(scorer)
ds.setScorerAlphaBeta(LM_ALPHA, LM_BETA)
while True: while True:
try: try: