|
|
|
@ -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
|
|
|
|
|
Alex Stoken
5 years ago