Glenn Jocher
4 years ago
1 changed files with 3 additions and 2 deletions
+ 3
- 2
train.py
View File
| 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) | |||||
| tb_writer.add_histogram('classes', c, 0) | |||||
| if tb_writer: | |||||
| 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) |
Loading…