Merge pull request #384 from jancio/master

Log hyperparameters in tensorboard

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