Syntax fixes
This commit is contained in:
1 parent
9b7386f603
commit
c8152c81a6
1 file changed
+2
-2
@@ -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
|
||||
|
||||
Reference in new issue
Block a user