From 17cf9f4a07d9a95c35b4644e6f6cc205740d02fa Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 11 Sep 2019 22:21:39 +0200 Subject: [PATCH] updates --- train.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/train.py b/train.py index 24165fd9..e3d4eceb 100644 --- a/train.py +++ b/train.py @@ -397,6 +397,12 @@ if __name__ == '__main__': device = torch_utils.select_device(opt.device, apex=mixed_precision) tb_writer = None + if opt.prebias: + train() # transfer-learn yolo biases for 1 epoch + create_backbone('weights/last.pt') # saved results as backbone.pt + opt.weights = 'weights/backbone.pt' # assign backbone + opt.prebias = False # disable prebias + if not opt.evolve: # Train normally try: # Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/ @@ -406,12 +412,6 @@ if __name__ == '__main__': except: pass - if opt.prebias: - train() # transfer-learn yolo biases for 1 epoch - create_backbone('weights/last.pt') # saved results as backbone.pt - opt.weights = 'weights/backbone.pt' # assign backbone - opt.prebias = False # disable prebias - train() # train normally else: # Evolve hyperparameters (optional)