Skip to content

Commit

Permalink
Fix references to AbstractTrainer.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 452815564
  • Loading branch information
MarkDaoust authored and copybara-github committed Jun 3, 2022
1 parent 0a1387b commit dfbed06
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions site/en/guide/model_garden/index.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit dfbed06

Please sign in to comment.