| 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 | |