Add save_dir to plot_lr_scheduler and plot_labels

Set save_dir = log_dir in train.py
This commit is contained in:
Alex Stoken committed 2020-06-17 16:03:18 -05:00
1 parent 3b2b330872
commit 945307beba
3 files changed
+6 -6

No files matched your search

+2 -2
View File
@@ -8,8 +8,8 @@
# train and val datasets (image directory or *.txt file with image paths)
train: ../coco128/images/train2017/
val: ../coco128/images/train2017/
train: C:/Users/astoken/projects/yolov5/data/coco/images/train2017
val: C:/Users/astoken/projects/yolov5/data/coco/images/train2017
# number of classes
nc: 80