Increase plot_labels() speed (#1736)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-12-18 18:05:38 -08:00
1 parent 49abc722fc
commit 685d601308
2 files changed
+10 -18

No files matched your search

+1 -1
View File
@@ -205,7 +205,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
# cf = torch.bincount(c.long(), minlength=nc) + 1. # frequency
# model._initialize_biases(cf.to(device))
if plots:
Thread(target=plot_labels, args=(labels, save_dir, loggers), daemon=True).start()
plot_labels(labels, save_dir, loggers)
if tb_writer:
tb_writer.add_histogram('classes', c, 0)