Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

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

Browse Source 
Ensures weights/ dir exists
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-06-13 12:54:43 -07:00
2 parents 099e6f5ebd c3d4d321d3
commit 19e68e8a7b
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1
View File
@@ -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'