diff --git a/keras_nlp/models/distil_bert/distil_bert_classifier.py b/keras_nlp/models/distil_bert/distil_bert_classifier.py index 770bf5e02b..42de1cee83 100644 --- a/keras_nlp/models/distil_bert/distil_bert_classifier.py +++ b/keras_nlp/models/distil_bert/distil_bert_classifier.py @@ -137,6 +137,7 @@ class DistilBertClassifier(Task): num_classes=4, ) classifier.fit(x=features, y=labels, batch_size=2) + ``` """ def __init__(