Improve documentation for custom training loops

[gachombamuthiri]

Improve documentation for custom training loops

[tilakrayal]

@gachombamuthiri,
Have you got the chance to have a look at this official documents where those are referring to the Custom training loop with Keras and MultiWorkerMirroredStrategy, Custom training and how to write the training loop from scratch with the examples. Could you please refer to the same and let me know if you are looking for the same.

https://www.tensorflow.org/tutorials/distribute/multi_worker_with_ctl
https://www.tensorflow.org/tutorials/customization/custom_training_walkthrough
https://www.tensorflow.org/guide/keras/writing_a_training_loop_from_scratch

Thank you!

[github-actions]

This issue is stale because it has been open for 7 days with no activity. It will be closed if no further activity occurs. Thank you.