提交 95f0a56d authored 作者: Alex Stoken's avatar Alex Stoken

Bug fix to get_latest_run when recent run is named

上级 52bac22f
......@@ -39,7 +39,7 @@ def init_seeds(seed=0):
def get_latest_run(search_dir = './runs'):
# get path to most recent 'last.pt' in run dirs
# assumes most recently saved 'last.pt' is the desired weights to --resume from
last_list = glob.glob(f'{search_dir}/**/last.pt', recursive=True)
last_list = glob.glob(f'{search_dir}/**/last*.pt', recursive=True)
latest = max(last_list, key = os.path.getctime)
return latest
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论