Skip to content

Commit 467c57b

Browse files
committed
add prefix option to generate checkpoint file name
1 parent 0b1eb77 commit 467c57b

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/lightning/pytorch/callbacks/model_checkpoint.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -591,7 +591,11 @@ def _format_checkpoint_name(
591591
return filename
592592

593593
def format_checkpoint_name(
594-
self, metrics: dict[str, Tensor], filename: Optional[str] = None, ver: Optional[int] = None
594+
self,
595+
metrics: dict[str, Tensor],
596+
filename: Optional[str] = None,
597+
prefix: Optional[str] = None,
598+
ver: Optional[int] = None,
595599
) -> str:
596600
"""Generate a filename according to the defined template.
597601
@@ -623,7 +627,9 @@ def format_checkpoint_name(
623627
624628
"""
625629
filename = filename or self.filename
626-
filename = self._format_checkpoint_name(filename, metrics, auto_insert_metric_name=self.auto_insert_metric_name)
630+
filename = self._format_checkpoint_name(
631+
filename, metrics, prefix, auto_insert_metric_name=self.auto_insert_metric_name
632+
)
627633

628634
if ver is not None:
629635
filename = self.CHECKPOINT_JOIN_CHAR.join((filename, f"v{ver}"))

0 commit comments

Comments
 (0)