Skip to content

GitLab

Commit 746a927c authored by Amit Patankar's avatar Amit Patankar
Browse files

Setting the initial learning rate.

parent 1cd1999e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/resnet/keras/keras_cifar_main.py
View file @ 746a927c
...@@ -116,6 +116,7 @@ def learning_rate_schedule(current_epoch, current_batch, batches_per_epoch): ...@@ -116,6 +116,7 @@ def learning_rate_schedule(current_epoch, current_batch, batches_per_epoch):
# return learning_rate # return learning_rate
epoch = current_epoch + float(current_batch) / batches_per_epoch epoch = current_epoch + float(current_batch) / batches_per_epoch
learning_rate = BASE_LEARNING_RATE
for mult, start_epoch in LR_SCHEDULE: for mult, start_epoch in LR_SCHEDULE:
if epoch >= start_epoch: if epoch >= start_epoch:
learning_rate = BASE_LEARNING_RATE * mult learning_rate = BASE_LEARNING_RATE * mult
......
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