Increase plot_labels() speed (#1736)
This commit is contained in:
1 parent
49abc722fc
commit
685d601308
2 files changed
+10
-18
No files matched your search
@@ -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)
|
||||
|
||||
|
||||
Reference in new issue
Block a user