update utils.py for support QDenseBN quant model save

google · Apr 2, 2022 · 93690e1 · 93690e1
1 parent 71207ce
commit 93690e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qkeras/utils.py b/qkeras/utils.py
@@ -198,7 +198,7 @@ def model_save_quantized_weights(model, filename=None):
         saved_weights[layer.name]["signs"] = signs
 
       if not any(isinstance(layer, t) for t in [
-          QConv2DBatchnorm, QDepthwiseConv2DBatchnorm]):
+          QConv2DBatchnorm, QDenseBatchnorm, QDepthwiseConv2DBatchnorm]):
         layer.set_weights(weights)
       else:
         print(layer.name, " conv and batchnorm weights cannot be seperately"