1 parent
f4197214aa
commit
08d4918d7f
2 files changed
+8
-3
No files matched your search
@@ -203,7 +203,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:
|
||||
plot_labels(labels, save_dir, loggers)
|
||||
plot_labels(labels, names, save_dir, loggers)
|
||||
if tb_writer:
|
||||
tb_writer.add_histogram('classes', c, 0)
|
||||
|
||||
|
||||
Reference in new issue
Block a user