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

Merge pull request #1990 from JRMeyer/evaluate_tflite 

Update evaluate_tflite.py script for Coqpit
This commit is contained in:
Reuben Morais 2021-10-30 10:18:56 +02:00 committed by GitHub
parent 65e66117e2 157ce340b6
commit 489e49f698
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
evaluate_tflite.py
View File

@ -10,10 +10,9 @@ import wave
from functools import partial from functools import partial
from multiprocessing import JoinableQueue, Manager, Process, cpu_count from multiprocessing import JoinableQueue, Manager, Process, cpu_count
import absl.app
import numpy as np import numpy as np
from coqui_stt_training.util.evaluate_tools import calculate_and_print_report from coqui_stt_training.util.evaluate_tools import calculate_and_print_report
from coqui_stt_training.util.flags import create_flags from coqui_stt_training.util.config import Config, initialize_globals_from_args
from six.moves import range, zip from six.moves import range, zip
from stt import Model from stt import Model
@ -61,6 +60,7 @@ def tflite_worker(model, scorer, queue_in, queue_out, gpu_mask):
def main(args): def main(args):
initialize_globals_from_args()
manager = Manager() manager = Manager()
work_todo = JoinableQueue() # this is where we are going to store input data work_todo = JoinableQueue() # this is where we are going to store input data
work_done = manager.Queue() # this where we are gonna push them out work_done = manager.Queue() # this where we are gonna push them out