Skip to content

Commit

Permalink
update release note
Browse files Browse the repository at this point in the history
  • Loading branch information
martinkim0 committed Apr 3, 2024
1 parent 13dd0b0 commit aaac7c4
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ is available in the [commit logs](https://github.com/scverse/scvi-tools/commits/
downloads of models from AWS S3 {pr}`2615`.
- Add support for `batch_key` in {meth}`scvi.model.CondSCVI.setup_anndata` {pr}`2626`.
- Add `load_best_on_end` argument to {class}`scvi.train.SaveCheckpoint` to load the best model
state at the end of training {pr}`xxxx`.
state at the end of training {pr}`2672`.

#### Changed

Expand Down
2 changes: 1 addition & 1 deletion scvi/train/_callbacks.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@


class SaveCheckpoint(ModelCheckpoint):
"""``EXPERIMENTAL`` Saves model checkpoints based on a monitored metric.
"""``BETA`` Saves model checkpoints based on a monitored metric.
Inherits from :class:`~lightning.pytorch.callbacks.ModelCheckpoint` and modifies the default
behavior to save the full model state instead of just the state dict. This is necessary for
Expand Down

0 comments on commit aaac7c4

Please sign in to comment.