Update default filename (#4500)

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Jeff Yang <ydcjeff@outlook.com>
This commit is contained in:
Rohit Gupta 2020-11-04 15:37:42 +05:30 committed by GitHub
parent fc78ffa622
commit db3cf2e1f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -366,7 +366,8 @@ class ModelCheckpoint(Callback):
) -> str:
if not filename:
# filename is not set, use default name
filename = "{epoch}-{step}"
filename = "{epoch}" + cls.CHECKPOINT_JOIN_CHAR + "{step}"
# check and parse user passed keys in the string
groups = re.findall(r"(\{.*?)[:\}]", filename)
if len(groups) >= 0: