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)