Skip to content
Snippets Groups Projects
Commit 282a3d84 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] exercise check_all_tensor_shapes_specified

parent 2390e812
No related branches found
No related tags found
No related merge requests found
......@@ -86,6 +86,7 @@ def test_modelwrapper():
lfc.load_state_dict(checkpoint["state_dict"])
bo.export_finn_onnx(lfc, (1, 1, 28, 28), export_onnx_path)
model = ModelWrapper(export_onnx_path)
assert model.check_all_tensor_shapes_specified() is False
inp_shape = model.get_tensor_shape("0")
assert inp_shape == [1, 1, 28, 28]
l0_weights = model.get_initializer("26")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment