Update eager/custom_training_walkthrough notebook to fit with collection.
Update eager/custom_training_walkthrough notebook to fit with collection.

tensorflow/docs,tensorflow/docs,tensorflow/docs-l10n,tensorflow/docs