tensorboard/focal loss reporting update

This commit is contained in:
Glenn Jocher 2020-04-06 15:45:18 -07:00
parent c7f93bae40
commit 05ae6e8499
1 changed files with 5 additions and 1 deletions

View File

@ -48,6 +48,10 @@ if f:
for k, v in zip(hyp.keys(), np.loadtxt(f[0])):
hyp[k] = v
# Print focal loss if gamma > 0
if hyp['fl_gamma']:
print('Using FocalLoss(gamma=%g)' % hyp['fl_gamma'])
def train():
cfg = opt.cfg
@ -412,7 +416,7 @@ if __name__ == '__main__':
from torch.utils.tensorboard import SummaryWriter
tb_writer = SummaryWriter()
print('Tensorboard started at http://localhost:6006/')
print("Tensorboard found. Run 'tensorboard --logdir=runs' to view at http://localhost:6006/")
except:
pass