Merge pull request #384 from jancio/master

Log hyperparameters in tensorboard
5 years ago · 1e94bcf3d2
parent a1c8406af3 120d40c06a
commit 1e94bcf3d2
1 changed files with 1 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -186,6 +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_histogram('classes', c, 0)

    # Check anchors