zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

comment tb_writer.add_hparams(hyp, {})

This commit is contained in:
Glenn Jocher 2020-07-15 10:38:32 -07:00
parent 95ef36bb74
commit 03489aaafb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -186,7 +186,7 @@ def train(hyp):
# model._initialize_biases(cf.to(device)) # model._initialize_biases(cf.to(device))
plot_labels(labels, save_dir=log_dir) plot_labels(labels, save_dir=log_dir)
if tb_writer: if tb_writer:
tb_writer.add_hparams(hyp, {}) # tb_writer.add_hparams(hyp, {}) # causes duplicate https://github.com/ultralytics/yolov5/pull/384
tb_writer.add_histogram('classes', c, 0) tb_writer.add_histogram('classes', c, 0)
# Check anchors # Check anchors