update model_save_quantized_weights to include QDenseBN

google · Apr 2, 2022 · 71207ce · 71207ce
1 parent 6a6599f
commit 71207ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qkeras/utils.py b/qkeras/utils.py
@@ -138,7 +138,7 @@ def model_save_quantized_weights(model, filename=None):
       signs = []
 
       if any(isinstance(layer, t) for t in [
-          QConv2DBatchnorm, QDepthwiseConv2DBatchnorm]):
+          QConv2DBatchnorm, QDenseBatchnorm, QDepthwiseConv2DBatchnorm]):
         qs = layer.get_quantizers()
         ws = layer.get_folded_weights()
       elif any(isinstance(layer, t) for t in [QSimpleRNN, QLSTM, QGRU]):