Fix bug in run_pretrained_models for large models (#1327)

Signed-off-by: Tom Wildenhain <tomwi@microsoft.com>
onnx · Feb 11, 2021 · fea121d · fea121d
1 parent c551408
commit fea121d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/run_pretrained_models.py b/tests/run_pretrained_models.py
@@ -438,6 +438,7 @@ def run_tflite():
                 tf_results = list(tf_results_d.values())
             else:
                 tf_results = [tf_results_d[output] for output in self.structured_outputs]
+            tf_results = [tf_res.numpy() for tf_res in tf_results]
             if self.perf:
                 logger.info("Running TF perf")
                 start = time.time()