We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c551408 commit fea121dCopy full SHA for fea121d
tests/run_pretrained_models.py
@@ -438,6 +438,7 @@ def run_tflite():
438
tf_results = list(tf_results_d.values())
439
else:
440
tf_results = [tf_results_d[output] for output in self.structured_outputs]
441
+ tf_results = [tf_res.numpy() for tf_res in tf_results]
442
if self.perf:
443
logger.info("Running TF perf")
444
start = time.time()
0 commit comments