Merge pull request #2259 from vaharoni:tutorial_text_fix

PiperOrigin-RevId: 563950730
tensorflow · Sep 9, 2023 · 377cb52 · 377cb52
2 parents 6c7e49a + 8822d90
commit 377cb52
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/site/en/tutorials/load_data/text.ipynb b/site/en/tutorials/load_data/text.ipynb
@@ -689,8 +689,7 @@
       },
       "outputs": [],
       "source": [
-        "# `vocab_size` is `VOCAB_SIZE + 1` since `0` is used additionally for padding.\n",
-        "int_model = create_model(vocab_size=VOCAB_SIZE + 1, num_labels=4)\n",
+        "int_model = create_model(vocab_size=VOCAB_SIZE, num_labels=4)\n",
         "int_model.compile(\n",
         "    loss=losses.SparseCategoricalCrossentropy(from_logits=True),\n",
         "    optimizer='adam',\n",