From 10871e747a0a25e84c86639c1beaa2949a9ba093 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicol=C3=B2=20Ghielmetti?= <nicolo.ghielmetti@gmail.com>
Date: Wed, 25 Oct 2023 11:58:53 +0200
Subject: [PATCH] check if  exists and in case use it

---
 qkeras/qlayers.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/qkeras/qlayers.py b/qkeras/qlayers.py
index 298a77a3..f5cff710 100644
--- a/qkeras/qlayers.py
+++ b/qkeras/qlayers.py
@@ -103,7 +103,10 @@ def __call__(self, shape, dtype=None):
 
     max_x = np.max(abs(x))
     std_x = np.std(x)
-    delta = self.quantizer.max() * 2**-self.quantizer.bits
+    if hasattr(self.quantizer, 'delta') and callable(getattr(self.quantizer, 'delta')):
+      delta = self.quantizer.delta()
+    else:
+      delta = self.quantizer.max() * 2**-self.quantizer.bits
 
     # delta is the minimum resolution of the number system.
     # we want to make sure we have enough values.