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

updates

This commit is contained in:
Glenn Jocher 2020-03-04 16:33:14 -08:00
parent 4a5159710f
commit 8b6c8a5318
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

@ -215,7 +215,7 @@ def train():
# Prebias # Prebias
if prebias: if prebias:
ne = 3 # number of prebias epochs ne = max(round(30 / nb), 3) # number of prebias epochs
ps = np.interp(epoch, [0, ne], [0.1, hyp['lr0'] * 2]), \ ps = np.interp(epoch, [0, ne], [0.1, hyp['lr0'] * 2]), \
np.interp(epoch, [0, ne], [0.9, hyp['momentum']]) # prebias settings (lr=0.1, momentum=0.9) np.interp(epoch, [0, ne], [0.9, hyp['momentum']]) # prebias settings (lr=0.1, momentum=0.9)
if epoch == ne: if epoch == ne: