We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 71207ce commit 93690e1Copy full SHA for 93690e1
qkeras/utils.py
@@ -198,7 +198,7 @@ def model_save_quantized_weights(model, filename=None):
198
saved_weights[layer.name]["signs"] = signs
199
200
if not any(isinstance(layer, t) for t in [
201
- QConv2DBatchnorm, QDepthwiseConv2DBatchnorm]):
+ QConv2DBatchnorm, QDenseBatchnorm, QDepthwiseConv2DBatchnorm]):
202
layer.set_weights(weights)
203
else:
204
print(layer.name, " conv and batchnorm weights cannot be seperately"
0 commit comments