updates

2019-08-23 15:37:25 +02:00 · 2019-08-23 15:37:25 +02:00 · 7593dedc4c
parent d279aa0021
commit 7593dedc4c
1 changed files with 1 additions and 1 deletions
--- 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)