We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 476e371 commit 12e4fc1Copy full SHA for 12e4fc1
qkeras/utils.py
@@ -383,7 +383,7 @@ def model_save_quantized_weights(model, filename=None):
383
if has_scale:
384
saved_weights[layer.name]["scales"] = scales
385
if not any(isinstance(layer, t) for t in [
386
- QConv2DBatchnorm, QDepthwiseConv2DBatchnorm]):
+ QConv2DBatchnorm, QDenseBatchnorm, QDepthwiseConv2DBatchnorm]):
387
# Set layer weights in the format that software inference uses
388
layer.set_weights(weights)
389
else:
0 commit comments