Skip to content

Commit 79a13ef

Browse files
committed
consistent v2 weight reader behavior
1 parent 481a142 commit 79a13ef

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hls4ml/converters/keras_v3_to_hls.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def get_weights_data(self, layer_name, var_name):
168168
for w in layer.weights:
169169
if var_name in w.name:
170170
return np.array(w)
171-
raise ValueError(f"Variable {var_name} not found in layer {layer.name}")
171+
return None
172172

173173
reader = DummyReader()
174174
input_shapes = [list(t.shape) for t in inp_tensors]

0 commit comments

Comments
 (0)