Internal change

PiperOrigin-RevId: 368868424

Internal change
PiperOrigin-RevId: 368868424
548336e2 · A. Unique TensorFlower · 9cfbc813 · 548336e2
Commit 548336e2 authored Apr 16, 2021 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/colab/fine_tuning_bert.ipynb official/colab/fine_tuning_bert.ipynb +1 -1

No files found.
--- a/official/colab/fine_tuning_bert.ipynb
+++ b/official/colab/fine_tuning_bert.ipynb
@@ -960,7 +960,7 @@
      "source": [
        "result = bert_classifier(my_examples, training=False)\n",
        "\n",
-        "result = tf.argmax(result).numpy()\n",
+        "result = tf.argmax(result, axis=-1).numpy()\n",
        "result"
      ]
    },