!9943 Fix mass lr import

From: @linqingke
Reviewed-by: @c_34,@wuxuejian
Signed-off-by: @c_34
pull/9943/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 28de04fb00

@ -17,7 +17,9 @@ from math import ceil
import numpy as np import numpy as np
import mindspore.nn.learning_rate_schedule as lr_schedules import mindspore.nn.learning_rate_schedule as lr_schedules
from mindspore import Tensor
from mindspore.ops import operations as P
import mindspore.common.dtype as mstype
def square_root_schedule(lr, update_num, decay_start_step, def square_root_schedule(lr, update_num, decay_start_step,
warmup_steps=2000, warmup_steps=2000,

Loading…
Cancel
Save