tensorflow/text

GenerateText method have no attribute model

Opened this issue · 1 comments

GenerateText method in "image_captioning" in https://www.tensorflow.org/text/tutorials/image_captioning have no attribute model.
Screenshot 2024-03-17 at 3 58 51 PM

Fixing it by adding a attribute _model in GenerateText() function

super().__init__() 
    self._model = None 
Screenshot 2024-03-17 at 4 02 17 PM

This will leads to type cast issue in masked_loss() function
so casting it to int32.
def masked_loss(labels, preds): labels = tf.cast(labels, tf.int32)