astoken commited on
Commit
c8152c8
1 Parent(s): 9b7386f

Syntax fixes

Browse files
Files changed (1) hide show
  1. train.py +2 -2
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