Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "be937d4a201a96a537d645992ebfbe21f70cc493"
Commit 3570932e authored by Amit Patankar's avatar Amit Patankar
Browse files

Updating the base learning rate to 3.2

parent abdc7f11
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/resnet/keras/keras_cifar_main.py
View file @ 3570932e
...@@ -80,7 +80,7 @@ class TimeHistory(tf.keras.callbacks.Callback): ...@@ -80,7 +80,7 @@ class TimeHistory(tf.keras.callbacks.Callback):
LR_SCHEDULE = [ # (multiplier, epoch to start) tuples LR_SCHEDULE = [ # (multiplier, epoch to start) tuples
(1.0, 5), (0.1, 30), (0.01, 60), (0.001, 80) (1.0, 5), (0.1, 30), (0.01, 60), (0.001, 80)
] ]
BASE_LEARNING_RATE = 0.4 #0.128 BASE_LEARNING_RATE = 3.2 #0.128
def learning_rate_schedule(current_epoch, current_batch, batches_per_epoch): def learning_rate_schedule(current_epoch, current_batch, batches_per_epoch):
"""Handles linear scaling rule, gradual warmup, and LR decay. """Handles linear scaling rule, gradual warmup, and LR decay.
......
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