Skip to content

GitLab

Commit 81a19f62 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 368868424
parent e3704ce2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/colab/fine_tuning_bert.ipynb
View file @ 81a19f62
...@@ -960,7 +960,7 @@ ...@@ -960,7 +960,7 @@
"source": [ "source": [
"result = bert_classifier(my_examples, training=False)\n", "result = bert_classifier(my_examples, training=False)\n",
"\n", "\n",
"result = tf.argmax(result).numpy()\n", "result = tf.argmax(result, axis=-1).numpy()\n",
"result" "result"
] ]
}, },
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment