comment tb_writer.add_hparams(hyp, {})

2020-07-15 10:38:32 -07:00 · 2020-07-15 10:38:32 -07:00 · 03489aaafb
parent 95ef36bb74
commit 03489aaafb
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -186,7 +186,7 @@ def train(hyp):
    # model._initialize_biases(cf.to(device))
    plot_labels(labels, save_dir=log_dir)
    if tb_writer:
-        tb_writer.add_hparams(hyp, {})
+        # tb_writer.add_hparams(hyp, {})  # causes duplicate https://github.com/ultralytics/yolov5/pull/384
        tb_writer.add_histogram('classes', c, 0)

    # Check anchors