From 9b0aa3eb4a602e39212cd4a2f72b3807f434dfc9 Mon Sep 17 00:00:00 2001 From: yvetteliu Date: Thu, 28 May 2020 15:23:28 +0800 Subject: [PATCH] Modify the importing path of BERT after its src has been moved. Fix an error. --- tests/perf_test/bert/test_bert_train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/perf_test/bert/test_bert_train.py b/tests/perf_test/bert/test_bert_train.py index 39803d68600..c2bb88b0477 100644 --- 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