diff --git a/train.py b/train.py
index 9175d4cf..6422cfe3 100644
--- a/train.py
+++ b/train.py
@@ -121,7 +121,7 @@ def train():
         start_epoch = chkpt['epoch'] + 1
         del chkpt
 
-    elif weights.endswith('.weights'):  # darknet format
+    else:  # darknet format
         # possible weights are 'yolov3.weights', 'yolov3-tiny.conv.15',  'darknet53.conv.74' etc.
         cutoff = load_darknet_weights(model, weights)