未验证 提交 25ccc63c 编写于 作者: Y Yuan Gao 提交者: GitHub

Update mobilenet_ssd_fluid.py

Update optimizer
上级 46b5c460
...@@ -240,7 +240,6 @@ def train(train_file_list, ...@@ -240,7 +240,6 @@ def train(train_file_list,
optimizer = fluid.optimizer.Momentum( optimizer = fluid.optimizer.Momentum(
learning_rate=fluid.learning_rate_decay.exponential_decay( learning_rate=fluid.learning_rate_decay.exponential_decay(
learning_rate=learning_rate, learning_rate=learning_rate,
global_step=global_step,
decay_steps=40000, decay_steps=40000,
decay_rate=0.1, decay_rate=0.1,
staircase=True), staircase=True),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册