!9943 Fix mass lr import

From: @linqingke Reviewed-by: @c_34,@wuxuejian Signed-off-by: @c_34
2020-12-15 14:44:44 +08:00 · 2020-12-15 14:44:44 +08:00 · 28de04fb00
parent c3e7bbb4c2 f26003e723
commit 28de04fb00
1 changed files with 3 additions and 1 deletions
--- a/model_zoo/official/nlp/mass/src/utils/lr_scheduler.py
+++ b/model_zoo/official/nlp/mass/src/utils/lr_scheduler.py
@ -17,7 +17,9 @@ from math import ceil
 import numpy as np

 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,
                         warmup_steps=2000,