Glenn Jocher
GitHub
4 years ago
1 changed files with 1 additions and 0 deletions
+ 1
- 0
train.py
View File
| @@ -186,6 +186,7 @@ def train(hyp): | |||
| # model._initialize_biases(cf.to(device)) | |||
| plot_labels(labels, save_dir=log_dir) | |||
| if tb_writer: | |||
| tb_writer.add_hparams(hyp, {}) | |||
| tb_writer.add_histogram('classes', c, 0) | |||
| # Check anchors | |||
Loading…