Changed to use float16 instead of int16 (#873)

ishii-norimi · web-flow · commit 7fcdf41acd93 · 2024-06-17T20:54:28.000+09:00
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)

Original file line number	Diff line number	Diff line change
`@@ -53,7 +53,7 @@`
`53`	`53`	`name="c",`
`54`	`54`	`data_type=data_type,`
`55`	`55`	`dims=(2, 2),`
`56`		`- vals=np.array(vals, dtype=np.int16).tobytes(),`
	`56`	`+ vals=np.array(vals, dtype=np.float16).tobytes(),`
`57`	`57`	`raw=True,`
`58`	`58`	`)`
`59`	`59`	`node = onnx.helper.make_node("Constant", inputs=[], outputs=["y"], value=C_init)`