Syntax fixes

5 years ago · c8152c81a6
parent 9b7386f603
commit c8152c81a6
1 changed files with 2 additions and 2 deletions
--- 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