Move results.txt from weights/ to log_dir

5 years ago · 3b2b330872
parent ade023cff2
commit 3b2b330872
1 changed files with 1 additions and 1 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 = wdir + 'results.txt'
+    results_file = log_dir + 'results.txt'

    epochs = opt.epochs  # 300
    batch_size = opt.batch_size  # 64