Skip to content

Commit

Permalink
fix mr ref (#5257)
Browse files Browse the repository at this point in the history
Co-authored-by: Thomas Kunwar <[email protected]>
  • Loading branch information
dberenbaum and yathomasi authored Jun 5, 2024
1 parent 44f53ae commit c1cd862
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions content/docs/dvclive/ml-frameworks/pytorch-lightning.md
Original file line number Diff line number Diff line change
Expand Up @@ -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]
<abbr>model registry</abbr> or automation scenarios).

- Save updates to the checkpoints directory at the end of training:

Expand Down Expand Up @@ -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

0 comments on commit c1cd862

Please sign in to comment.