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

updates

This commit is contained in:
Glenn Jocher 2019-08-23 16:04:45 +02:00
parent 7f8318e680
commit 06e274f7e4
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

@ -128,7 +128,7 @@ def train():
if opt.transfer: # transfer learning if opt.transfer: # transfer learning
nf = int(model.module_defs[model.yolo_layers[0] - 1]['filters']) # yolo layer size (i.e. 255) nf = int(model.module_defs[model.yolo_layers[0] - 1]['filters']) # yolo layer size (i.e. 255)
for x in optimizer.param_groups: for x in optimizer.param_groups: # lower parameter count can handle more aggressive training hyps
x['lr'] *= 10 x['lr'] *= 10
x['momentum'] *= 0.9 x['momentum'] *= 0.9