apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher 2019-02-10 21:34:15 +01:00
parent 917f9dd248
commit 715c4575bf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -111,7 +111,7 @@ def train(
for g in optimizer.param_groups: for g in optimizer.param_groups:
g['lr'] = lr g['lr'] = lr
# Freeze darknet53.conv.74 layers for first epoch # Freeze darknet53.conv.74 for first epoch
if freeze_backbone: if freeze_backbone:
if epoch == 0: if epoch == 0:
for i, (name, p) in enumerate(model.named_parameters()): for i, (name, p) in enumerate(model.named_parameters()):