From f26003e723a23fda74f535560b928cef85e46096 Mon Sep 17 00:00:00 2001 From: linqingke Date: Mon, 14 Dec 2020 21:12:54 +0800 Subject: [PATCH] fix mass lr import. --- model_zoo/official/nlp/mass/src/utils/lr_scheduler.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/model_zoo/official/nlp/mass/src/utils/lr_scheduler.py b/model_zoo/official/nlp/mass/src/utils/lr_scheduler.py index 16607678e5..fe5edb7975 100644 --- 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,