From 7593dedc4c61c04f4d79887477115679d93f8a2a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 23 Aug 2019 15:37:25 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 6422cfe3..0362a7a9 100644 --- a/train.py +++ b/train.py @@ -121,7 +121,7 @@ def train(): start_epoch = chkpt['epoch'] + 1 del chkpt - else: # darknet format + elif len(weights) > 0: # darknet format # possible weights are 'yolov3.weights', 'yolov3-tiny.conv.15', 'darknet53.conv.74' etc. cutoff = load_darknet_weights(model, weights)