Skip to content

Commit 80173fb

Browse files
Fix inference with tf1
1 parent ee09857 commit 80173fb

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

bioimageio/core/prediction_pipeline/_model_adapters/_tensorflow_model_adapter.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,8 @@ def _forward_tf(self, *input_tensors):
7272

7373
# run prediction
7474
res = sess.run(dict(zip(out_names, out_tensors)), dict(zip(in_tensors, input_tensors)))
75+
# from dict to list of tensors
76+
res = [res[out] for out in out_names]
7577

7678
return res
7779

0 commit comments

Comments
 (0)