ViT-B-32__openai/textual/ Runs with emulator now.

2024-12-01 16:42:53 +00:00
parent 7fddf282cf
commit bc849e2e9f
3 changed files with 14 additions and 7 deletions
--- a/machine-learning/app/sessions/ort.py
+++ b/machine-learning/app/sessions/ort.py
@@ -46,6 +46,7 @@ class OrtSession:
        input_feed: dict[str, NDArray[np.float32]] | dict[str, NDArray[np.int32]],
        run_options: Any = None,
    ) -> list[NDArray[np.float32]]:
+        print(input_feed)
        outputs: list[NDArray[np.float32]] = self.session.run(output_names, input_feed, run_options)
        return outputs