mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Syntax fixes

Browse Source
This commit is contained in:
Alex Stoken 2020-06-17 16:32:13 -05:00
parent 9b7386f603
commit c8152c81a6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -50,7 +50,7 @@ def train(hyp):
os.makedirs(wdir, exist_ok=True) os.makedirs(wdir, exist_ok=True)
last = wdir + 'last.pt' last = wdir + 'last.pt'
best = wdir + 'best.pt' best = wdir + 'best.pt'
results_file = log_dir + 'results.txt' results_file = log_dir + os.sep + 'results.txt'
epochs = opt.epochs # 300 epochs = opt.epochs # 300
batch_size = opt.batch_size # 64 batch_size = opt.batch_size # 64
@ -303,7 +303,7 @@ def train(hyp):
model=ema.ema, model=ema.ema,
single_cls=opt.single_cls, single_cls=opt.single_cls,
dataloader=testloader, dataloader=testloader,
fast=epoch < epochs / 2 fast=epoch < epochs / 2,
save_dir=log_dir) save_dir=log_dir)
# Write # Write