From 38acc5f3c5ad6aa9960896caeb92fd54196b91a0 Mon Sep 17 00:00:00 2001 From: Janko Ondras Date: Tue, 14 Jul 2020 08:48:46 +0200 Subject: [PATCH] Fix img_size naming in hyperparameters logging --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 7b3494a..7669219 100644 --- a/train.py +++ b/train.py @@ -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, {})