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-25 01:58:12 +02:00
parent a85f7d967c
commit 991362df57
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -39,7 +39,7 @@ def train():
cfg = opt.cfg cfg = opt.cfg
data = opt.data data = opt.data
img_size = opt.img_size img_size = opt.img_size
epochs = 1 if opt.prebias else opt.epochs # 500200 batches at bs 16, 117263 images = 273 epochs epochs = 3 if opt.prebias else opt.epochs # 500200 batches at bs 16, 117263 images = 273 epochs
batch_size = opt.batch_size batch_size = opt.batch_size
accumulate = opt.accumulate # effective bs = batch_size * accumulate = 16 * 4 = 64 accumulate = opt.accumulate # effective bs = batch_size * accumulate = 16 * 4 = 64
weights = opt.weights # initial training weights weights = opt.weights # initial training weights
@ -110,7 +110,7 @@ def train():
for x in optimizer.param_groups: for x in optimizer.param_groups:
# lower param count allows more aggressive training settings: ~0.1 lr0, ~0.9 momentum # lower param count allows more aggressive training settings: ~0.1 lr0, ~0.9 momentum
x['lr'] *= 10 x['lr'] *= 100
x['momentum'] *= 0.9 x['momentum'] *= 0.9
for p in model.parameters(): for p in model.parameters():