Added missing external scorer

This commit is contained in:
Jedrzej Beniamin Orbik 2020-04-20 16:04:54 +02:00
parent bf3ebefd60
commit a4b08594eb

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):
os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_mask)
ds = Model(model)
ds.enableExternalScorer(scorer)
while True:
try: