zjc
/
yolov5
關註 1
收藏 0
複製 0
程式碼 問題管理 0 合併請求 0 版本發佈 0 Wiki Activity
瀏覽代碼

Ensures weights/ dir exists 

Allows train.py to be run outside of yolov5/ directory.
5.0
Lauritzen Kasper Primdal 4 年之前
父節點
099e6f5ebd
當前提交
c3d4d321d3
共有 1 個文件被更改，包括 1 次插入 和 0 次删除
分割檢視
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      train.py

+ 1
- 0
train.py 查看文件

@@ -20,6 +20,7 @@ except:
mixed_precision = False # not installed

wdir = 'weights' + os.sep # weights dir
os.makedirs(wdir, exist_ok=True)
last = wdir + 'last.pt'
best = wdir + 'best.pt'
results_file = 'results.txt'

Write Preview
Loading…
取消
儲存