Move results.txt from weights/ to log_dir
Browse files
train.py
CHANGED
|
@@ -50,7 +50,7 @@ def train(hyp):
|
|
| 50 |
os.makedirs(wdir, exist_ok=True)
|
| 51 |
last = wdir + 'last.pt'
|
| 52 |
best = wdir + 'best.pt'
|
| 53 |
-
results_file =
|
| 54 |
|
| 55 |
epochs = opt.epochs # 300
|
| 56 |
batch_size = opt.batch_size # 64
|
|
|
|
| 50 |
os.makedirs(wdir, exist_ok=True)
|
| 51 |
last = wdir + 'last.pt'
|
| 52 |
best = wdir + 'best.pt'
|
| 53 |
+
results_file = log_dir + 'results.txt'
|
| 54 |
|
| 55 |
epochs = opt.epochs # 300
|
| 56 |
batch_size = opt.batch_size # 64
|