updates

2019-02-10 21:34:15 +01:00 · 2019-02-10 21:34:15 +01:00 · 715c4575bf
commit 715c4575bf
parent 917f9dd248
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -111,7 +111,7 @@ def train(
        for g in optimizer.param_groups:
            g['lr'] = lr

-        # Freeze darknet53.conv.74 layers for first epoch
+        # Freeze darknet53.conv.74 for first epoch
        if freeze_backbone:
            if epoch == 0:
                for i, (name, p) in enumerate(model.named_parameters()):