Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2018-12-12 17:26:46 +01:00
parent b5a2747a6a
commit 3c95b5c104
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

@ -125,7 +125,7 @@ def train(
g['lr'] = lr g['lr'] = lr
# Freeze darknet53.conv.74 layers for first epoch # Freeze darknet53.conv.74 layers for first epoch
if freeze_backbone is not False: 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()):
if int(name.split('.')[1]) < 75: # if layer < 75 if int(name.split('.')[1]) < 75: # if layer < 75