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

Fix img_size naming in hyperparameters logging

This commit is contained in:
Janko Ondras 2020-07-14 08:48:46 +02:00
parent e558963244
commit 38acc5f3c5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -62,7 +62,7 @@ def train(hyp):
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']
tb_hparams_dict['img_size_train'], tb_hparams_dict['img_size_test'] = tb_hparams_dict['img_size']
del tb_hparams_dict['img_size']
tb_writer.add_hparams(tb_hparams_dict, {})