Changed to use float16 instead of int16 (#873)

ai-on-browser · Jun 17, 2024 · 7fcdf41 · 7fcdf41
1 parent e4d44fb
commit 7fcdf41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/lib/model/nns/onnx/utils.py b/tests/lib/model/nns/onnx/utils.py
@@ -53,7 +53,7 @@
         name="c",
         data_type=data_type,
         dims=(2, 2),
-        vals=np.array(vals, dtype=np.int16).tobytes(),
+        vals=np.array(vals, dtype=np.float16).tobytes(),
         raw=True,
     )
     node = onnx.helper.make_node("Constant", inputs=[], outputs=["y"], value=C_init)