File size: 400 Bytes
cea4a4b | 1 2 3 4 5 6 7 8 9 10 | def get_new_model_name(model_checkpoint: str, model_name: str = None) -> str:
if model_name is None:
old_version_number = int(model_checkpoint[-2:])
new_version_number = str(old_version_number + 1).zfill(2)
model_name = f"{model_checkpoint[:-2]}{new_version_number}"
elif not model_name[-2:].isnumeric():
model_name = model_name + "_00"
return model_name
|