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

updates

This commit is contained in:
Glenn Jocher 2019-04-02 19:19:02 +02:00
parent f527d30ccd
commit 9b32d3cb5d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -40,6 +40,8 @@ def train(
# Initialize model # Initialize model
model = Darknet(cfg, img_size).to(device) model = Darknet(cfg, img_size).to(device)
# for m in model.modules():
# weights_init_normal(m) # set weight distributions
# Optimizer # Optimizer
lr0 = 0.001 # initial learning rate lr0 = 0.001 # initial learning rate