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