Subclassing keras.Model optimizer errors

Question

I\'m trying to subclass keras.Model in order to do text segmentation. More specifically, I\'m trying to train a model that outputs a probability for each sentence,