Skip to content

Y
yolov5
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
上级 099e6f5e c3d4d321
隐藏空白字符变更
内嵌 并排
正在显示 包含 1 行增加 和 0 行删除
train.py
浏览文件 @ 19e68e8a
...@@ -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 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论