mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-06-16 09:59:13 +02:00
Predictor for multi-input models: present as list instead of tuple…
(because TF-Serving expects that and cannot cast)
This commit is contained in:
parent
4e7e1c06b9
commit
38fe4d33ad
1 changed files with 2 additions and 2 deletions
|
|
@ -184,7 +184,7 @@ class Predictor(mp.context.SpawnProcess):
|
|||
else:
|
||||
data.append(stack.enter_context(ndarray_shared(shared_data)))
|
||||
if multi_input:
|
||||
data = tuple(np.concatenate(data0)
|
||||
data = list(np.concatenate(data0)
|
||||
for data0 in zip(*data))
|
||||
else:
|
||||
data = np.concatenate(data)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue