This commit is contained in:
Eren Gölge 2021-02-18 18:24:03 +00:00
parent adaeec57ec
commit 547bfc4ce9
1 changed files with 1 additions and 1 deletions

View File

@ -94,7 +94,7 @@ def get_last_checkpoint(path):
match = re.search(f"{key}_([0-9]+)", file_name)
if match is not None:
model_num = int(match.groups()[0])
if model_num > last_model_num or last_model_num is None:
if last_model_num is None or model_num > last_model_num:
last_model_num = model_num
last_model = file_name