Spaces:
Build error
Build error
File size: 741 Bytes
65d9523 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
import os
import glob
from pathlib import Path
def increment_name(path):
"increase save directory's id"
path = Path(path)
sep = ''
if path.exists():
path, suffix = (path.with_suffix(''), path.suffix) if path.is_file() else (path, '')
for n in range(1, 9999):
p = f'{path}{sep}{n}{suffix}'
if not os.path.exists(p):
break
path = Path(p)
return path
def find_latest_checkpoint(search_dir='.'):
# Find the most recent saved checkpoint in search_dir
checkpoint_list = glob.glob(f'{search_dir}/**/last*.pt', recursive=True)
return max(checkpoint_list, key=os.path.getctime) if checkpoint_list else ''
|