Spaces:

xiang-wuu
/

yolov5

Runtime error

Janko Ondras commited on Jul 14, 2020

Commit

38acc5f

1 Parent(s): e558963

Fix img_size naming in hyperparameters logging

Files changed (1) hide show

train.py CHANGED Viewed

@@ -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']
         del tb_hparams_dict['img_size']
         tb_writer.add_hparams(tb_hparams_dict, {})

     if tb_writer:
         tb_hparams_dict = hyp
         tb_hparams_dict.update(vars(opt))
+        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, {})