提交 03489aaa authored 作者: Glenn Jocher's avatar Glenn Jocher

comment tb_writer.add_hparams(hyp, {})

上级 95ef36bb
...@@ -186,7 +186,7 @@ def train(hyp): ...@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论