From 715c4575bf8f17c914ee3172bee5c3585e045f40 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 10 Feb 2019 21:34:15 +0100 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 69eb7bf7..293ceee1 100644 --- 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()):