updates

2018-12-12 17:26:46 +01:00 · 2018-12-12 17:26:46 +01:00 · 3c95b5c104
parent b5a2747a6a
commit 3c95b5c104
1 changed files with 1 additions and 1 deletions
--- 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