Merge pull request #384 from jancio/master
Browse files
train.py
CHANGED
@@ -186,6 +186,7 @@ def train(hyp):
|
|
186 |
# model._initialize_biases(cf.to(device))
|
187 |
plot_labels(labels, save_dir=log_dir)
|
188 |
if tb_writer:
|
|
|
189 |
tb_writer.add_histogram('classes', c, 0)
|
190 |
|
191 |
# Check anchors
|
|
|
186 |
# model._initialize_biases(cf.to(device))
|
187 |
plot_labels(labels, save_dir=log_dir)
|
188 |
if tb_writer:
|
189 |
+
tb_writer.add_hparams(hyp, {})
|
190 |
tb_writer.add_histogram('classes', c, 0)
|
191 |
|
192 |
# Check anchors
|