Merge pull request #384 from jancio/master

Log hyperparameters in tensorboard
ultralytics · Jul 14, 2020 · 1e94bcf · 1e94bcf
2 parents a1c8406 + 120d40c
commit 1e94bcf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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