Skip to content

Commit

Permalink
update symlink option for save_last in ModelCheckpoint (#11319)
Browse files Browse the repository at this point in the history
Signed-off-by: paul-gibbons <[email protected]>
  • Loading branch information
paul-gibbons authored Nov 19, 2024
1 parent 7d32537 commit a273ddd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nemo/lightning/pytorch/callbacks/model_checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
import shutil
from datetime import timedelta
from pathlib import Path
from typing import Any, Dict, Iterable, List, Optional, Union
from typing import Any, Dict, Iterable, List, Literal, Optional, Union

import lightning.pytorch
import torch
Expand Down Expand Up @@ -63,7 +63,7 @@ def __init__(
self,
monitor: Optional[str] = "val_loss",
verbose: bool = True,
save_last: Optional[bool] = True,
save_last: Optional[Union[bool, Literal["link"]]] = True,
save_top_k: int = 3,
save_weights_only: bool = False, ## TODO: check support
mode: str = "min",
Expand Down

0 comments on commit a273ddd

Please sign in to comment.