#!/usr/bin/env python # coding: utf-8 from tensorflow.keras.datasets import mnist from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Activation, Flatten, Input from tensorflow.keras.utils import to_categorical from qkeras.qlayers import QDense from qkeras.quantizers import quantized_bits (train_data, train_labels), (test_data, test_labels) = mnist.load_data() train_data = train_data.astype('float32') / 255.0 test_images = test_data.astype('float32') / 255.0 train_labels = to_categorical(train_labels) test_labels = to_categorical(test_labels) model = Sequential([ Input(name="Input", shape=(28, 28)), Flatten(name="Flatten"), QDense(32, name="QDense1", kernel_quantizer=quantized_bits(2, 0), bias_quantizer=quantized_bits(2, 0)), Activation("softmax", name="Softmax") ]) model.compile() model.fit(train_data, train_labels) 

Traceback (most recent call last): File "/home/me/wtf/test.py", line 22, in <module> model.fit(train_data, train_labels) File "/home/me/.pyenv/versions/3.11.8/lib/python3.11/site-packages/keras/src/utils/traceback_utils.py", line 122, in error_handler raise e.with_traceback(filtered_tb) from None File "/home/me/.pyenv/versions/3.11.8/lib/python3.11/site-packages/qkeras/qlayers.py", line 629, in call quantized_kernel = self.kernel_quantizer_internal(self.kernel) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/me/.pyenv/versions/3.11.8/lib/python3.11/site-packages/qkeras/quantizers.py", line 558, in __call__ x = K.cast_to_floatx(x) ^^^^^^^^^^^^^^^^^^^ NotImplementedError: Exception encountered when calling QDense.call(). numpy() is only available when eager execution is enabled. Arguments received by QDense.call(): • inputs=tf.Tensor(shape=(32, 784), dtype=float32)