From dfbed06c3f5f82431f6bec8e1b2fe2c7b33b92a2 Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Fri, 3 Jun 2022 12:07:16 -0700 Subject: [PATCH] Fix references to `AbstractTrainer`. PiperOrigin-RevId: 452815564 --- site/en/guide/model_garden/index.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/site/en/guide/model_garden/index.md b/site/en/guide/model_garden/index.md index c08ec950a95..57b022a62d1 100644 --- a/site/en/guide/model_garden/index.md +++ b/site/en/guide/model_garden/index.md @@ -124,8 +124,9 @@ Alternatively, you can use the high-level Keras `Model.fit`. However, if your model is complex and your training loop requires more flexible control or customization, then you should use Orbit. You can define most of your -training loop by extending Orbit's `AbstractTrainerclass`. Learn more about the -Orbit tool in the [Orbit API documentation](../../api_docs/python/orbit). +training loop by the `orbit.AbstractTrainer` or `orbit.StandardTrainer` class. +Learn more about the Orbit tool in the +[Orbit API documentation](../../api_docs/python/orbit). Note: You can use the Keras API to do what Orbit does, but you must override the TensorFlow `train_step` function or use callbacks like ModelCheckpoint or