Add a link to the guide.

Fixes tensorflow#40402 PiperOrigin-RevId: 365826243 Change-Id: I84f502e9aa1df65a3ad180931f942c10bdd1754c
Marbck · Mar 30, 2021 · ce8db8a · ce8db8a
1 parent da877df
commit ce8db8a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow/python/keras/engine/training.py b/tensorflow/python/keras/engine/training.py
@@ -766,6 +766,8 @@ def train_step(self, data):
     """The logic for one training step.
 
     This method can be overridden to support custom training logic.
+    For concrete examples of how to override this method see
+    [Customizing what happends in fit](https://www.tensorflow.org/guide/keras/customizing_what_happens_in_fit).
     This method is called by `Model.make_train_function`.
 
     This method should contain the mathematical logic for one step of training.