diff --git a/src/HGQ/layers/base.py b/src/HGQ/layers/base.py index e573880..cbdb144 100644 --- a/src/HGQ/layers/base.py +++ b/src/HGQ/layers/base.py @@ -43,7 +43,7 @@ def __init__(self, kq_conf=None, paq_conf=None, beta=0., **kwargs): "kernel quantizer config" self.paq_config = paq_conf or get_default_paq_conf() "pre-activation quantizer config" - self.beta = tf.Variable(beta, dtype=tf.float32, name='beta') + self.beta = tf.Variable(beta, dtype=tf.float32, name='beta', trainable=False) "BOPs-regularization strength" self.record_minmax = False self._has_last_layer = False