Merge pull request #2027 from sushreebarsa:patch-17

copybara-github · copybara-github · commit d5cda5e803f0 · 2022-02-17T21:17:25.000-08:00
PiperOrigin-RevId: 429474125
diff --git a/site/en/tutorials/customization/custom_training_walkthrough.ipynb b/site/en/tutorials/customization/custom_training_walkthrough.ipynb
@@ -464,7 +464,7 @@
         "id": "uRZmchElo481"
       },
       "source": [
-        "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:"
+        "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:"
       ]
     },
     {

Original file line number	Diff line number	Diff line change
`@@ -464,7 +464,7 @@`
`464`	`464`	`"id": "uRZmchElo481"`
`465`	`465`	`},`
`466`	`466`	`"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:"
`468`	`468`	`]`
`469`	`469`	`},`
`470`	`470`	`{`