I am having a terrible time analyzing a model using Cube AI.
When I analyze the TFlite model, I get an internal error.
When I analyze the same model saved in Keras, I get the error below. However, the model trains and works perfectly for inference in tensorflow.
I have decorated the classes, for example, this way:
@tf.keras.utils.register_keras_serializable(package="Custom", name="PatchEmbedding"), but it still doesn't work.
Can anyone help me out?