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

Log hyperparameters in tensorboard 

Log both hyperparameters and command line options in tensorboard.
This commit is contained in:
Janko Ondras 2020-07-13 12:55:43 +02:00
parent e169edfcf5
commit e558963244
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
train.py
View File

@ -58,6 +58,14 @@ def train(hyp):
with open(Path(log_dir) / 'opt.yaml', 'w') as f: with open(Path(log_dir) / 'opt.yaml', 'w') as f:
yaml.dump(vars(opt), f, sort_keys=False) yaml.dump(vars(opt), f, sort_keys=False)
# Log hyperparameters in tensorboard
if tb_writer:
tb_hparams_dict = hyp
tb_hparams_dict.update(vars(opt))
tb_hparams_dict['img_size_w'], tb_hparams_dict['img_size_h'] = tb_hparams_dict['img_size']
del tb_hparams_dict['img_size']
tb_writer.add_hparams(tb_hparams_dict, {})
epochs = opt.epochs # 300 epochs = opt.epochs # 300
batch_size = opt.batch_size # 64 batch_size = opt.batch_size # 64
weights = opt.weights # initial training weights weights = opt.weights # initial training weights