From c1cd862b0f78e5b662c3b6c514e7dc1af99da058 Mon Sep 17 00:00:00 2001 From: Dave Berenbaum Date: Wed, 5 Jun 2024 10:18:41 -0400 Subject: [PATCH] fix mr ref (#5257) Co-authored-by: Thomas Kunwar <20840228+yathomasi@users.noreply.github.com> --- content/docs/dvclive/ml-frameworks/pytorch-lightning.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/content/docs/dvclive/ml-frameworks/pytorch-lightning.md b/content/docs/dvclive/ml-frameworks/pytorch-lightning.md index f9561728e7..8ac01ff4a6 100644 --- a/content/docs/dvclive/ml-frameworks/pytorch-lightning.md +++ b/content/docs/dvclive/ml-frameworks/pytorch-lightning.md @@ -93,8 +93,8 @@ checkpointing at all as described in the Use `log_model` to save the checkpoints (it will use `Live.log_artifact()` internally to save those). At the end of training, DVCLive will copy the [`best_model_path`][`ModelCheckpoint`] to the `dvclive/artifacts` directory and -annotate it with name `best` (for example, to be consumed in [DVC Studio model -registry] or automation scenarios). +annotate it with name `best` (for example, to be consumed in [DVC Studio] +model registry or automation scenarios). - Save updates to the checkpoints directory at the end of training: @@ -188,3 +188,4 @@ dvclive/metrics/train/epoch/metric.tsv [`live`]: /doc/dvclive/live [`ModelCheckpoint`]: https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.callbacks.ModelCheckpoint.html +[DVC Studio]: https://studio.iterative.ai