Syntax fixes
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 = log_dir + 'results.txt'
|
54 |
|
55 |
epochs = opt.epochs # 300
|
56 |
batch_size = opt.batch_size # 64
|
@@ -303,7 +303,7 @@ def train(hyp):
|
|
303 |
model=ema.ema,
|
304 |
single_cls=opt.single_cls,
|
305 |
dataloader=testloader,
|
306 |
-
fast=epoch < epochs / 2
|
307 |
save_dir=log_dir)
|
308 |
|
309 |
# Write
|
|
|
50 |
os.makedirs(wdir, exist_ok=True)
|
51 |
last = wdir + 'last.pt'
|
52 |
best = wdir + 'best.pt'
|
53 |
+
results_file = log_dir + os.sep + 'results.txt'
|
54 |
|
55 |
epochs = opt.epochs # 300
|
56 |
batch_size = opt.batch_size # 64
|
|
|
303 |
model=ema.ema,
|
304 |
single_cls=opt.single_cls,
|
305 |
dataloader=testloader,
|
306 |
+
fast=epoch < epochs / 2,
|
307 |
save_dir=log_dir)
|
308 |
|
309 |
# Write
|