From 2856af5036de44188a18928368dbd726b42bbacb Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 21 Mar 2019 12:11:08 +0200
Subject: [PATCH] updates

---
 train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train.py b/train.py
index 572ff973..f37bae87 100644
--- a/train.py
+++ b/train.py
@@ -64,9 +64,9 @@ def train(
     if torch.cuda.device_count() > 1:
         model = nn.DataParallel(model)
 
-    # # Transfer learning (train only YOLO layers)
-    for i, (name, p) in enumerate(model.named_parameters()):
-        p.requires_grad = True if (p.shape[0] == 255) else False
+    # Transfer learning (train only YOLO layers)
+    # for i, (name, p) in enumerate(model.named_parameters()):
+    #     p.requires_grad = True if (p.shape[0] == 255) else False
 
     # Set scheduler
     # scheduler = torch.optim.lr_scheduler.MultiStepLR(optimizer, milestones=[54, 61], gamma=0.1)