| import pathlib | |
| import shutil | |
| ckpt_paths = pathlib.Path("./").glob("**/model.ckpt") | |
| for ckpt_path in ckpt_paths: | |
| print(ckpt_path) | |
| shutil.move( | |
| ckpt_path, | |
| f"{ckpt_path.parent/ ckpt_path.parent.stem}.ckpt" | |
| ) | |
| print(f"{ckpt_path.parent/ ckpt_path.parent.stem}.ckpt") | |