Browse Source

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

Merge pull request !1584 from lyvette/master
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
4da9fe083b
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      tests/perf_test/bert/test_bert_train.py

+ 1
- 2
tests/perf_test/bert/test_bert_train.py View File

@@ -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


Loading…
Cancel
Save