From 885d8bcf40f23c982c322f617f79d38b3d406971 Mon Sep 17 00:00:00 2001 From: Ram Rachum Date: Fri, 9 Aug 2024 19:02:38 +0300 Subject: [PATCH] Fix link to orbax documentation --- docs/guides/training_techniques/use_checkpointing.ipynb | 2 +- docs/guides/training_techniques/use_checkpointing.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/guides/training_techniques/use_checkpointing.ipynb b/docs/guides/training_techniques/use_checkpointing.ipynb index 0937a636d..8d2346776 100644 --- a/docs/guides/training_techniques/use_checkpointing.ipynb +++ b/docs/guides/training_techniques/use_checkpointing.ipynb @@ -15,7 +15,7 @@ "* Support for various array types and storage formats\n", "* Asynchronous saving to reduce training wait time\n", "* Versioning and automatic bookkeeping of past checkpoints\n", - "* Flexible [`transformations`](https://github.com/google/orbax/blob/main/docs/checkpoint.md#transformations) to tweak and load old checkpoints\n", + "* Flexible [`transformations`](https://orbax.readthedocs.io/en/latest/transformations.html) to tweak and load old checkpoints\n", "* [`jax.sharding`](https://jax.readthedocs.io/en/latest/notebooks/Distributed_arrays_and_automatic_parallelization.html)-based API to save and load in multi-host scenarios\n", "\n", "---\n", diff --git a/docs/guides/training_techniques/use_checkpointing.md b/docs/guides/training_techniques/use_checkpointing.md index b24fa4f3c..8df64047a 100644 --- a/docs/guides/training_techniques/use_checkpointing.md +++ b/docs/guides/training_techniques/use_checkpointing.md @@ -19,7 +19,7 @@ Orbax provides a variety of features for saving and loading model data, which yo * Support for various array types and storage formats * Asynchronous saving to reduce training wait time * Versioning and automatic bookkeeping of past checkpoints -* Flexible [`transformations`](https://github.com/google/orbax/blob/main/docs/checkpoint.md#transformations) to tweak and load old checkpoints +* Flexible [`transformations`](https://orbax.readthedocs.io/en/latest/transformations.html) to tweak and load old checkpoints * [`jax.sharding`](https://jax.readthedocs.io/en/latest/notebooks/Distributed_arrays_and_automatic_parallelization.html)-based API to save and load in multi-host scenarios ---