From fdd5afa2290c789f5bc5f86febc56af7b768f83d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 8 Aug 2019 22:37:54 +0200 Subject: [PATCH] updates --- train.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/train.py b/train.py index ec776be4..d55a02bb 100644 --- a/train.py +++ b/train.py @@ -359,18 +359,15 @@ if __name__ == '__main__': opt = parser.parse_args() print(opt) - - if not opt.evolve: # Train normally # Tensorboard support, - # start with "tensorboard --logdir=runs" - # go to localhost:6006 + # start with "tensorboard --logdir=runs" then go to localhost:6006 tensorboard_support = True - if version_to_tuple(torch.__version__)>= version_to_tuple("1.1.0"): - try: + if version_to_tuple(torch.__version__) >= version_to_tuple("1.1.0"): + try: from torch.utils.tensorboard import SummaryWriter + tb_train_name = time.time() - print('here') tb_writer = SummaryWriter('runs/{}'.format(tb_train_name)) except: tensorboard_support = False @@ -381,7 +378,7 @@ if __name__ == '__main__': epochs=opt.epochs, batch_size=opt.batch_size, accumulate=opt.accumulate, - write_to_tensorboard=tensorboard_results) + write_to_tensorboard=tensorboard_support) else: # Evolve hyperparameters (optional) opt.notest = True # only test final epoch