Skip to content

Commit

Permalink
Update orbax_upgrade_guide.rst
Browse files Browse the repository at this point in the history
  • Loading branch information
kaushaladiti-2802 committed Jul 2, 2024
1 parent 4cc1329 commit baad060
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ Then, you can call ``orbax.checkpoint.AsyncCheckpointer.wait_until_finished()``

For more details, read the `checkpoint guide <https://flax.readthedocs.io/en/latest/guides/training_techniques/use_checkpointing.html#asynchronized-checkpointing>`_.

You can also use Orbax AsyncCheckpointer with Flax APIs through async manager. Async manager internally calls wait_until_finished(). This solution is not actively mainted and the recommedation is to use Orbax async checkpointing.
You can also use Orbax AsyncCheckpointer with Flax APIs through async manager. Async manager internally calls wait_until_finished(). This solution is not actively maintained and the recommedation is to use Orbax async checkpointing.

For example:

Expand Down

0 comments on commit baad060

Please sign in to comment.