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

Update tensorboard logging

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-10-26 01:08:33 +01:00
1 parent 7f1640695b
commit 453acdec67
3 files changed
+9 -8
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

train.py
+4 -4
View File
@@ -291,11 +291,11 @@ def train(hyp, opt, device, tb_writer=None):
# Plot
if ni < 3:
f = str(log_dir / ('train_batch%g.jpg' % ni)) # filename
f = str(log_dir / f'train_batch{ni}.jpg') # filename
result = plot_images(images=imgs, targets=targets, paths=paths, fname=f)
if tb_writer and result is not None:
tb_writer.add_image(f, result, dataformats='HWC', global_step=epoch)
# tb_writer.add_graph(model, imgs) # add model to tensorboard
# if tb_writer and result is not None:
# tb_writer.add_image(f, result, dataformats='HWC', global_step=epoch)
# tb_writer.add_graph(model, imgs) # add model to tensorboard
# end batch ------------------------------------------------------------------------------------------------