From 3c95b5c104d906536cdae584f05d062846a62813 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 12 Dec 2018 17:26:46 +0100
Subject: [PATCH] updates

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

diff --git a/train.py b/train.py
index c232cd0e..d463a650 100644
--- a/train.py
+++ b/train.py
@@ -125,7 +125,7 @@ def train(
             g['lr'] = lr
 
         # Freeze darknet53.conv.74 layers for first epoch
-        if freeze_backbone is not False:
+        if freeze_backbone:
             if epoch == 0:
                 for i, (name, p) in enumerate(model.named_parameters()):
                     if int(name.split('.')[1]) < 75:  # if layer < 75