Commit
·
03489aa
1
Parent(s):
95ef36b
comment tb_writer.add_hparams(hyp, {})
Browse files
train.py
CHANGED
@@ -186,7 +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_hparams(hyp, {})
|
190 |
tb_writer.add_histogram('classes', c, 0)
|
191 |
|
192 |
# 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, {}) # causes duplicate https://github.com/ultralytics/yolov5/pull/384
|
190 |
tb_writer.add_histogram('classes', c, 0)
|
191 |
|
192 |
# Check anchors
|