From d279aa0021583ba26bfa1d37a5e38352cde55b36 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 23 Aug 2019 15:35:39 +0200
Subject: [PATCH] updates

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)