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

Merge pull request #2925 from Jendker/evaluate_tflite_fix

Browse Source 
Add missing external scorer
This commit is contained in:
lissyx 2020-04-21 15:34:29 +02:00 committed by GitHub
commit cc449e0443
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
evaluate_tflite.py
View File

@ -32,6 +32,7 @@ Then run with a TF Lite model, a scorer and a CSV test file
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.enableExternalScorer(scorer)
while True: while True:
try: try: