!1584 Modify the importing path of BERT after its src has been moved.

Merge pull request !1584 from lyvette/master
5 years ago · 4da9fe083b
parent 618b2c99c2 9b0aa3eb4a
commit 4da9fe083b
1 changed files with 1 additions and 2 deletions
--- a/tests/perf_test/bert/test_bert_train.py
+++ b/tests/perf_test/bert/test_bert_train.py
@ -22,8 +22,7 @@ import os
 import mindspore.common.dtype as mstype
 import mindspore.context as context
 from mindspore import Tensor
-from mindspore.model_zoo.Bert_NEZHA import BertConfig, BertNetworkWithLoss, BertTrainOneStepCell, \
-    BertTrainOneStepWithLossScaleCell
+from model_zoo.bert.src import BertConfig, BertNetworkWithLoss, BertTrainOneStepCell, BertTrainOneStepWithLossScaleCell
 from mindspore.nn.optim import AdamWeightDecayDynamicLR
 from mindspore.train.loss_scale_manager import DynamicLossScaleManager
 from ...dataset_mock import MindData