Commit c4a7f16b authored by Yuxin Wu's avatar Yuxin Wu

bug fix on resnet

parent dfacc168
......@@ -121,7 +121,6 @@ class Model(ModelDesc):
# weight decay on all W of fc layers
wd_w = tf.train.exponential_decay(1e-4, get_global_step_var(),
200000, 0.7, True)
wd_w = wd_w / tf.get_default_graph().get_tensor_by_name('learning_rate')
wd_cost = tf.mul(wd_w, regularize_cost('.*/W', tf.nn.l2_loss), name='l2_regularize_loss')
add_moving_summary(loss, wd_cost)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment