Update train.py

bubbliiiing · web-flow · commit e93fce2c20a1 · 2021-10-04T01:03:33.000+08:00
diff --git a/train.py b/train.py
@@ -212,7 +212,7 @@
         #------------------------------------#
         if Freeze_Train:
             for param in model.backbone.parameters():
-                param.requires_grad = False
+                param.requires_grad = True
 
         for epoch in range(start_epoch, end_epoch):
             fit_one_epoch(model_train, model, yolo_loss, loss_history, optimizer, epoch, 
