zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

evolution bug fix #346

This commit is contained in:
Glenn Jocher 2020-07-09 22:08:42 -07:00
parent 59beae98e8
commit 1c13e67b33
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
train.py
View File

@ -44,7 +44,7 @@ hyp = {'optimizer': 'SGD', # ['adam', 'SGD', None] if none, default is SGD
def train(hyp): def train(hyp):
print(f'Hyperparameters {hyp}') print(f'Hyperparameters {hyp}')
log_dir = tb_writer.log_dir # run directory log_dir = tb_writer.log_dir if tb_writer else 'runs/evolution' # run directory
wdir = str(Path(log_dir) / 'weights') + os.sep # weights directory wdir = str(Path(log_dir) / 'weights') + os.sep # weights directory
os.makedirs(wdir, exist_ok=True) os.makedirs(wdir, exist_ok=True)
@ -387,7 +387,10 @@ if __name__ == '__main__':
opt.weights = last if opt.resume and not opt.weights else opt.weights opt.weights = last if opt.resume and not opt.weights else opt.weights
opt.cfg = check_file(opt.cfg) # check file opt.cfg = check_file(opt.cfg) # check file
opt.data = check_file(opt.data) # check file opt.data = check_file(opt.data) # check file
opt.hyp = check_file(opt.hyp) if opt.hyp else '' # check file if opt.hyp: # update hyps
opt.hyp = check_file(opt.hyp) # check file
with open(opt.hyp) as f:
hyp.update(yaml.load(f, Loader=yaml.FullLoader)) # update hyps
print(opt) print(opt)
opt.img_size.extend([opt.img_size[-1]] * (2 - len(opt.img_size))) # extend to 2 sizes (train, test) opt.img_size.extend([opt.img_size[-1]] * (2 - len(opt.img_size))) # extend to 2 sizes (train, test)
device = torch_utils.select_device(opt.device, apex=mixed_precision, batch_size=opt.batch_size) device = torch_utils.select_device(opt.device, apex=mixed_precision, batch_size=opt.batch_size)
@ -396,12 +399,8 @@ if __name__ == '__main__':
# Train # Train
if not opt.evolve: if not opt.evolve:
print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
tb_writer = SummaryWriter(log_dir=increment_dir('runs/exp', opt.name)) tb_writer = SummaryWriter(log_dir=increment_dir('runs/exp', opt.name))
if opt.hyp: # update hyps print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
with open(opt.hyp) as f:
hyp.update(yaml.load(f, Loader=yaml.FullLoader))
train(hyp) train(hyp)
# Evolve hyperparameters (optional) # Evolve hyperparameters (optional)