Skip to content

Commit 93690e1

Browse files
authored
update utils.py for support QDenseBN quant model save
1 parent 71207ce commit 93690e1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

qkeras/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def model_save_quantized_weights(model, filename=None):
198198
saved_weights[layer.name]["signs"] = signs
199199

200200
if not any(isinstance(layer, t) for t in [
201-
QConv2DBatchnorm, QDepthwiseConv2DBatchnorm]):
201+
QConv2DBatchnorm, QDenseBatchnorm, QDepthwiseConv2DBatchnorm]):
202202
layer.set_weights(weights)
203203
else:
204204
print(layer.name, " conv and batchnorm weights cannot be seperately"

0 commit comments

Comments
 (0)