Merge pull request tensorflow#1306 from birkanatici:patch-1

PiperOrigin-RevId: 286917042
martbox3721 · Dec 23, 2019 · e1a844e · e1a844e
2 parents 72677ae + 5e06547
commit e1a844e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/site/en/tutorials/text/image_captioning.ipynb b/site/en/tutorials/text/image_captioning.ipynb
@@ -552,7 +552,7 @@
         "BUFFER_SIZE = 1000\n",
         "embedding_dim = 256\n",
         "units = 512\n",
-        "vocab_size = len(tokenizer.word_index) + 1\n",
+        "vocab_size = top_k + 1\n",
         "num_steps = len(img_name_train) // BATCH_SIZE\n",
         "# Shape of the vector extracted from InceptionV3 is (64, 2048)\n",
         "# These two variables represent that vector shape\n",