Skip to content

Commit fea121d

Browse files
Fix bug in run_pretrained_models for large models (#1327)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent c551408 commit fea121d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/run_pretrained_models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -438,6 +438,7 @@ def run_tflite():
438438
tf_results = list(tf_results_d.values())
439439
else:
440440
tf_results = [tf_results_d[output] for output in self.structured_outputs]
441+
tf_results = [tf_res.numpy() for tf_res in tf_results]
441442
if self.perf:
442443
logger.info("Running TF perf")
443444
start = time.time()

0 commit comments

Comments
 (0)