Unverified 提交 19e68e8a authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Merge pull request #51 from KPLauritzen/b-ensure-weights-dir-when-training

Ensures weights/ dir exists
...@@ -20,6 +20,7 @@ except: ...@@ -20,6 +20,7 @@ except:
mixed_precision = False # not installed mixed_precision = False # not installed
wdir = 'weights' + os.sep # weights dir wdir = 'weights' + os.sep # weights dir
os.makedirs(wdir, exist_ok=True)
last = wdir + 'last.pt' last = wdir + 'last.pt'
best = wdir + 'best.pt' best = wdir + 'best.pt'
results_file = 'results.txt' results_file = 'results.txt'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论