Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

tb_writer bug fix

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-06-16 16:05:28 -07:00
1 parent 4052603e1f
commit 8db51c7002
1 file changed
+3 -2
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+3 -2
View File
@@ -195,8 +195,9 @@ def train(hyp):
c = torch.tensor(labels[:, 0]) # classes c = torch.tensor(labels[:, 0]) # classes
# cf = torch.bincount(c.long(), minlength=nc) + 1. # cf = torch.bincount(c.long(), minlength=nc) + 1.
# model._initialize_biases(cf.to(device)) # model._initialize_biases(cf.to(device))
plot_labels(labels) if tb_writer:
tb_writer.add_histogram('classes', c, 0) plot_labels(labels)
tb_writer.add_histogram('classes', c, 0)
# Check anchors # Check anchors
check_anchors(dataset, model=model, thr=hyp['anchor_t'], imgsz=imgsz) check_anchors(dataset, model=model, thr=hyp['anchor_t'], imgsz=imgsz)