Skip to content

Commit f09bb16

Browse files
authored
Update custom_training_walkthrough.ipynb
Changed the naming from 'tf.argmax' to 'tf.math.argmax' in the documentation.
1 parent 4ac1b76 commit f09bb16

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

site/en/tutorials/customization/custom_training_walkthrough.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -464,7 +464,7 @@
464464
"id": "uRZmchElo481"
465465
},
466466
"source": [
467-
"Taking the `tf.argmax` across classes gives us the predicted class index. But, the model hasn't been trained yet, so these aren't good predictions:"
467+
"Taking the `tf.math.argmax` across classes gives us the predicted class index. But, the model hasn't been trained yet, so these aren't good predictions:"
468468
]
469469
},
470470
{

0 commit comments

Comments
 (0)