use standed ”import“ in cases

tests/st/auto_monad/test_auto_monad_mindtester.py

@@ -21,7 +21,7 @@ from mindspore.nn import Cell
 from mindspore import context, Tensor
 from mindspore.common.parameter import Parameter
 from mindspore.common.initializer import initializer
-from mindspore.train.model import Model
+from mindspore.train import Model
 from mindspore.ops.composite import GradOperation
 from mindspore.common import ParameterTuple
 from tests.security_utils import security_off_wrap

tests/st/auto_parallel/cell_shard.py

@@ -26,8 +26,7 @@ from mindspore.communication.management import init
 from mindspore.nn.loss.loss import LossBase
 from mindspore.nn.optim.momentum import Momentum
 from mindspore.ops import operations as P
-from mindspore.train.callback import Callback
-from mindspore.train.model import Model
+from mindspore.train import Callback, Model
 from mindspore.context import ParallelMode
 import mindspore.dataset as ds
 

tests/st/auto_parallel/multifieldembeddinglookup_parallel.py

@@ -23,8 +23,8 @@ from mindspore.nn import MultiFieldEmbeddingLookup as embedding
 from mindspore import Tensor
 from mindspore import context
 from mindspore.train import Model
-from mindspore.train.callback import CheckpointConfig
-from mindspore.train.callback import ModelCheckpoint
+from mindspore.train import CheckpointConfig
+from mindspore.train import ModelCheckpoint
 from mindspore.train.serialization import load_checkpoint
 from mindspore.train.serialization import load_param_into_net
 from mindspore.communication.management import init

tests/st/auto_parallel/optimizer_parallel.py

@@ -27,8 +27,8 @@ from mindspore.nn import Flatten
 from mindspore.nn import Momentum
 import mindspore.ops.operations as P
 from mindspore.train.serialization import load_param_into_net
-from mindspore.train.callback import CheckpointConfig
-from mindspore.train.callback import ModelCheckpoint
+from mindspore.train import CheckpointConfig
+from mindspore.train import ModelCheckpoint
 from mindspore.train.serialization import load_checkpoint
 
 from mindspore.nn import SoftmaxCrossEntropyWithLogits

tests/st/auto_parallel/parallel_strategy_search.py

@@ -26,8 +26,8 @@ from mindspore.nn import Dense
 from mindspore.nn import Softmax
 import mindspore.ops.operations as P
 from mindspore.train.serialization import load_param_into_net
-from mindspore.train.callback import CheckpointConfig
-from mindspore.train.callback import ModelCheckpoint
+from mindspore.train import CheckpointConfig
+from mindspore.train import ModelCheckpoint
 from mindspore.train.serialization import load_checkpoint
 from mindspore.nn import Momentum
 from mindspore.nn import SoftmaxCrossEntropyWithLogits

tests/st/auto_parallel/resnet50_expand_loss.py

@@ -27,8 +27,8 @@ from mindspore.nn.loss.loss import LossBase
 from mindspore.nn.optim.momentum import Momentum
 from mindspore.ops import operations as P
 from mindspore.parallel import set_algo_parameters
-from mindspore.train.callback import Callback
-from mindspore.train.model import Model
+from mindspore.train import Callback
+from mindspore.train import Model
 from mindspore.context import ParallelMode
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")

tests/st/auto_parallel/soft_entropy_loss_expand_parallel.py

@@ -27,8 +27,7 @@ from mindspore.nn import Cell
 from mindspore.nn.optim.momentum import Momentum
 from mindspore.ops import functional as F
 from mindspore.ops import operations as P
-from mindspore.train import Model
-from mindspore.train.callback import Callback
+from mindspore.train import Model, Callback
 
 np.set_printoptions(threshold=np.inf)
 device_num = 2

tests/st/broadcast/lenet_broadcast_auto_parallel.py

@@ -20,9 +20,8 @@ import numpy as np
 import mindspore.communication.management as distributedTool
 import mindspore.nn as nn
 from mindspore import context
-from mindspore.train.metrics import Accuracy
-from mindspore.train import Model
-from mindspore.train.callback import LossMonitor, TimeMonitor
+from mindspore.train import Accuracy
+from mindspore.train import Model, LossMonitor, TimeMonitor
 from tests.models.official.cv.lenet.src.dataset import create_dataset
 from tests.models.official.cv.lenet.src.lenet import LeNet5
 

tests/st/dataset/test_gpu_reset.py

@@ -16,10 +16,9 @@ import pytest
 
 from mindspore.ops import operations as P
 import mindspore.nn as nn
-from mindspore.train import Model
+from mindspore.train import Model, Callback
 from mindspore.common import set_seed
 import mindspore.dataset as ds
-from mindspore.train.callback import Callback
 from mindspore import log as logger
 
 set_seed(1)

tests/st/dump/test_dynamic_shape.py

@@ -33,7 +33,7 @@ from mindspore.nn import Dense
 from mindspore.nn import SoftmaxCrossEntropyWithLogits
 from mindspore.nn import WithLossCell
 from mindspore import dataset as ds
-from mindspore import Model
+from mindspore.train import Model
 from dump_test_utils import generate_dump_json, generate_statistic_dump_json, check_dump_structure
 from tests.security_utils import security_off_wrap
 

tests/st/dynamic_shape/test_dcn_dynamic_gpu.py

@@ -19,8 +19,7 @@ import pytest
 from mindspore import ops, nn, Tensor, Parameter, ParameterTuple, context, set_seed
 from mindspore.common.initializer import initializer, XavierUniform
 import mindspore.dataset as ds
-from mindspore.train.callback import Callback
-from mindspore.train import Model
+from mindspore.train import Callback, Model
 from mindspore.common import dtype as mstype
 import mindspore as ms
 

tests/st/dynamic_shape/test_dynamic_asr.py

@@ -35,8 +35,7 @@ from mindspore.nn.wrap.grad_reducer import DistributedGradReducer
 from mindspore.ops import composite as C, functional as F
 from mindspore.ops.functional import stop_gradient
 from mindspore.parallel._utils import _get_parallel_mode, _get_device_num, _get_gradients_mean
-from mindspore.train import Model
-from mindspore.train.callback._callback import Callback
+from mindspore.train import Model, Callback
 
 logging.basicConfig(level=logging.DEBUG, format="%(asctime)s %(levelname)s %(message)s")
 

tests/st/dynamic_shape/test_dynamic_transformer.py

@@ -21,9 +21,8 @@ import pytest
 import mindspore as ms
 from mindspore.common.tensor import Tensor
 from mindspore.nn.optim import Adam
-from mindspore.train.model import Model
 from mindspore.train.loss_scale_manager import DynamicLossScaleManager
-from mindspore.train.callback import Callback, TimeMonitor
+from mindspore.train import Callback, TimeMonitor, Model
 from mindspore.common import set_seed
 import mindspore.dataset as de
 from transformer.transformer_for_train import TransformerNetworkWithLoss, TransformerTrainOneStepWithLossScaleCell

tests/st/dynamic_shape/test_dynamic_wenet_ascend.py

@@ -39,8 +39,7 @@ from mindspore.nn.wrap.loss_scale import DynamicLossScaleUpdateCell
 from mindspore.nn.wrap.grad_reducer import DistributedGradReducer
 from mindspore.nn.optim import Adam
 
-from mindspore.train import Model
-from mindspore.train.callback._callback import Callback
+from mindspore.train import Model, Callback
 from mindspore import context, ParameterTuple, set_seed
 from mindspore.context import ParallelMode
 from mindspore.communication.management import get_group_size

tests/st/dynamic_shape/yolov5_dynamic/run_yolov5_dynamic.py

@@ -18,7 +18,7 @@ import numpy as np
 from mindspore.train import Model
 from mindspore import Tensor
 import mindspore.dataset as ds
-from mindspore.train.callback import LossMonitor
+from mindspore.train import LossMonitor
 import mindspore as ms
 import mindspore.nn as nn
 import mindspore.communication as comm

tests/st/export_and_load/test_ckpt.py

@@ -22,7 +22,7 @@ import mindspore.dataset as ds
 import mindspore.dataset.transforms as C
 import mindspore.dataset.vision as CV
 import mindspore.nn as nn
-from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, Callback
+from mindspore.train import ModelCheckpoint, CheckpointConfig, Callback
 from mindspore import load_checkpoint
 from mindspore.common import dtype as mstype
 from mindspore.dataset.vision import Inter

tests/st/frontend_compile_cache/run_lenet_ps.py

@@ -23,9 +23,8 @@ import mindspore.dataset.vision as CV
 import mindspore.nn as nn
 from mindspore.common import dtype as mstype
 from mindspore.dataset.vision import Inter
-from mindspore.train.metrics import Accuracy
-from mindspore.train import Model
-from mindspore.train.callback import LossMonitor
+from mindspore.train import Accuracy
+from mindspore.train import Model, LossMonitor
 from mindspore.common.initializer import TruncatedNormal
 from mindspore.communication.management import init
 

tests/st/map_parameter/network/src/model.py

@@ -28,9 +28,8 @@ from mindspore.experimental import MapParameter
 from mindspore.nn import Cell, Flatten, Dense
 from mindspore.nn import SoftmaxCrossEntropyWithLogits
 from mindspore.nn import Adam
-from mindspore.train import Model
-from mindspore.train.callback import CheckpointConfig, ModelCheckpoint
-from mindspore.train.metrics import Accuracy
+from mindspore.train import Model, CheckpointConfig, ModelCheckpoint
+from mindspore.train import Accuracy
 from mindspore.common import set_seed
 
 

tests/st/mem_reuse/resnet_cifar_memreuse.py

@@ -29,8 +29,7 @@ from mindspore.communication.management import init
 from mindspore.nn.optim.momentum import Momentum
 from mindspore.ops import functional as F
 from mindspore.ops import operations as P
-from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor
-from mindspore.train.model import Model
+from mindspore.train import ModelCheckpoint, CheckpointConfig, LossMonitor, Model
 from mindspore.context import ParallelMode
 
 random.seed(1)

tests/st/mem_reuse/resnet_cifar_normal.py

@@ -29,8 +29,7 @@ from mindspore.communication.management import init
 from mindspore.nn.optim.momentum import Momentum
 from mindspore.ops import functional as F
 from mindspore.ops import operations as P
-from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor
-from mindspore.train.model import Model
+from mindspore.train import ModelCheckpoint, CheckpointConfig, LossMonitor, Model
 from mindspore.context import ParallelMode
 
 random.seed(1)

tests/st/mix_precision/test_mix_precision.py

@@ -24,7 +24,7 @@ from mindspore import amp
 from mindspore import Tensor
 from mindspore import context
 from mindspore.train.loss_scale_manager import FixedLossScaleManager
-from mindspore.train.model import Model
+from mindspore.train import Model
 from utils import FakeData
 from utils import allclose_nparray
 from utils import FakeDataInitMode

tests/st/model_zoo_tests/DeepFM/src/callback.py

@@ -16,7 +16,7 @@
 Defined callback for DeepFM.
 """
 import time
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 
 
 def add_write(file_path, out_str):

tests/st/model_zoo_tests/DeepFM/src/deepfm.py

@@ -22,10 +22,10 @@ from mindspore.ops import composite as C
 from mindspore.ops import operations as P
 from mindspore.nn import Dropout
 from mindspore.nn.optim import Adam
-from mindspore.train.metrics import Metric
+from mindspore.train import Metric
 from mindspore import nn, Tensor, ParameterTuple, Parameter
 from mindspore.common.initializer import Uniform, initializer
-from mindspore.train.callback import ModelCheckpoint, CheckpointConfig
+from mindspore.train import ModelCheckpoint, CheckpointConfig
 from mindspore.parallel._utils import _get_device_num, _get_parallel_mode, _get_gradients_mean
 from mindspore.context import ParallelMode
 from mindspore.nn.wrap.grad_reducer import DistributedGradReducer

tests/st/model_zoo_tests/DeepFM/test_deepfm.py

@@ -17,7 +17,7 @@ import os
 import pytest
 
 from mindspore import context
-from mindspore.train.model import Model
+from mindspore.train import Model
 from mindspore.common import set_seed
 
 from src.deepfm import ModelBuilder, AUCMetric

tests/st/model_zoo_tests/deeplabv3/train_one_epoch_with_loss.py

@@ -17,9 +17,8 @@ import argparse
 import time
 from mindspore import context
 from mindspore.nn.optim.momentum import Momentum
-from mindspore import Model
 from mindspore.train.serialization import load_checkpoint, load_param_into_net
-from mindspore.train.callback import Callback
+from mindspore.train import Callback, Model
 from src.md_dataset import create_dataset
 from src.losses import OhemLoss
 from src.deeplabv3 import deeplabv3_resnet50

tests/st/model_zoo_tests/transformer/test_transformer.py

@@ -21,9 +21,8 @@ import numpy as np
 import mindspore.common.dtype as mstype
 from mindspore.common.tensor import Tensor
 from mindspore.nn.optim import Adam
-from mindspore.train.model import Model
+from mindspore.train import Model, Callback
 from mindspore.train.loss_scale_manager import DynamicLossScaleManager
-from mindspore.train.callback import Callback
 import mindspore.dataset as ds
 import mindspore.dataset.transforms as deC
 from mindspore import context

tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/callbacks.py

@@ -15,7 +15,7 @@
 callbacks
 """
 import time
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 from mindspore import context
 from mindspore.context import ParallelMode
 from mindspore.communication.management import get_rank

tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/train_and_test_multinpu_ci.py

@@ -18,7 +18,7 @@
 import os
 import sys
 from mindspore import Model, context
-from mindspore.train.callback import TimeMonitor
+from mindspore.train import TimeMonitor
 from mindspore.context import ParallelMode
 from mindspore.communication.management import get_rank, get_group_size, init
 from mindspore.nn.wrap.cell_wrapper import VirtualDatasetCellTriple

tests/st/model_zoo_tests/wide_and_deep/train_and_test_multinpu_ci_data_parallel.py

@@ -18,7 +18,7 @@ import os
 import sys
 import numpy as np
 from mindspore import Model, context
-from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, TimeMonitor
+from mindspore.train import ModelCheckpoint, CheckpointConfig, TimeMonitor
 from mindspore.context import ParallelMode
 from mindspore.communication.management import get_rank, get_group_size, init
 

tests/st/model_zoo_tests/yolov3/test_yolov3.py

@@ -28,9 +28,8 @@ import pytest
 import numpy as np
 import mindspore.nn as nn
 from mindspore import context, Tensor
-from mindspore.train import Model
+from mindspore.train import Model, Callback
 from mindspore.common.initializer import initializer
-from mindspore.train.callback import Callback
 
 from src.yolov3 import yolov3_resnet18, YoloWithLossCell, TrainingWrapper
 from src.dataset import create_yolo_dataset

tests/st/model_zoo_tests/yolov3_darknet53/test_yolov3_darknet53.py

@@ -27,7 +27,7 @@ import numpy as np
 
 from mindspore import context, Tensor
 from mindspore.common.initializer import initializer
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 from mindspore.context import ParallelMode
 from mindspore.nn.optim.momentum import Momentum
 import mindspore as ms

tests/st/networks/models/bert/bert_performance/src/utils.py

@@ -26,9 +26,8 @@ from mindspore import log as logger
 from mindspore.common import dtype as mstype
 from mindspore.common.tensor import Tensor
 from mindspore.nn.learning_rate_schedule import LearningRateSchedule, PolynomialDecayLR, WarmUpLR
-from mindspore.train.metrics import Metric
 from mindspore.ops import operations as P
-from mindspore.train.callback import Callback
+from mindspore.train import Callback, Metric
 
 
 class CrossEntropyCalculation(nn.Cell):

tests/st/networks/models/bert/bert_performance/test_bert_tdt_ge.py

@@ -25,8 +25,7 @@ from mindspore import log as logger
 from mindspore.ops import operations as P
 from mindspore.common.tensor import Tensor
 from mindspore.nn.optim import Lamb
-from mindspore.train.callback import Callback
-from mindspore.train.model import Model
+from mindspore.train import Callback, Model
 import mindspore.nn.learning_rate_schedule as lr_schedules
 from tests.models.official.nlp.bert.src.bert_for_pre_training import BertNetworkWithLoss
 from tests.models.official.nlp.bert.src.bert_for_pre_training import BertTrainOneStepCell

tests/st/networks/models/bert/bert_performance/test_bert_tdt_lossscale.py

@@ -27,9 +27,9 @@ from mindspore import log as logger
 from mindspore.ops import operations as P
 from mindspore.common.tensor import Tensor
 from mindspore.nn.optim import Lamb
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 from mindspore.train.loss_scale_manager import DynamicLossScaleManager
-from mindspore.train.model import Model
+from mindspore.train import Model
 import mindspore.nn.learning_rate_schedule as lr_schedules
 from tests.models.official.nlp.bert.src.bert_for_pre_training import BertNetworkWithLoss
 from tests.models.official.nlp.bert.src.bert_for_pre_training import BertTrainOneStepWithLossScaleCell

tests/st/networks/models/bert/bert_performance/test_bert_thor.py

@@ -25,12 +25,11 @@ import mindspore.common.dtype as mstype
 import mindspore.communication.management as D
 from mindspore import context
 from mindspore import log as logger
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 from mindspore.context import ParallelMode
 from mindspore.train.serialization import load_checkpoint, load_param_into_net
 from mindspore.nn.optim import thor
-from mindspore.train.model import Model
-from mindspore.train.train_thor import ConvertModelUtils
+from mindspore.train import Model, ConvertModelUtils
 import mindspore.dataset.transforms as C
 
 from tests.st.networks.models.bert.bert_performance.src.bert_for_pre_training import BertNetworkWithLoss, \

tests/st/networks/models/bert/bert_precision/test_bert_tdt_lossscale.py

@@ -27,9 +27,8 @@ from mindspore import log as logger
 from mindspore.ops import operations as P
 from mindspore.common.tensor import Tensor
 from mindspore.nn.optim import Lamb
-from mindspore.train.callback import Callback
 from mindspore.train.loss_scale_manager import DynamicLossScaleManager
-from mindspore.train.model import Model
+from mindspore.train import Model, Callback
 import mindspore.nn.learning_rate_schedule as lr_schedules
 from tests.models.official.nlp.bert.src.bert_for_pre_training import BertNetworkWithLoss
 from tests.models.official.nlp.bert.src.bert_for_pre_training import BertTrainOneStepWithLossScaleCell

tests/st/networks/models/deeplabv3/src/miou_precision.py

@@ -14,7 +14,7 @@
 # ============================================================================
 """mIou."""
 import numpy as np
-from mindspore.train.metrics import Metric
+from mindspore.train import Metric
 
 
 def confuse_matrix(target, pred, n):

tests/st/networks/models/deeplabv3/test_deeplabv3.py

@@ -20,7 +20,7 @@ import numpy as np
 from mindspore import context, Tensor
 from mindspore.nn.optim.momentum import Momentum
 from mindspore import Model
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 from src.md_dataset import create_dataset
 from src.losses import OhemLoss
 from src.deeplabv3 import deeplabv3_resnet50

tests/st/networks/models/resnet50/src/callback.py

@@ -16,7 +16,7 @@
 import time
 import numpy as np
 import mindspore as ms
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 
 
 class LossGet(Callback):

tests/st/networks/models/resnet50/test_resnet50_imagenet.py

@@ -22,9 +22,9 @@ import numpy as np
 from mindspore import context
 from mindspore.common.tensor import Tensor
 import mindspore.communication.management as D
-from mindspore.train.model import Model
+from mindspore.train import Model
 from mindspore.context import ParallelMode
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 from mindspore.train.loss_scale_manager import FixedLossScaleManager
 import mindspore.nn as nn
 import mindspore.dataset as ds

tests/st/networks/models/resnet50/train_resnet50_thor.py

@@ -17,7 +17,7 @@ import os
 import numpy as np
 import mindspore as ms
 from mindspore import nn
-from mindspore.train.train_thor import ConvertModelUtils
+from mindspore.train import ConvertModelUtils
 from tests.st.networks.models.resnet50.src.callback import LossGet
 from tests.st.networks.models.resnet50.src_thor.config import config as thor_config
 from tests.st.networks.models.resnet50.src_thor.dataset import create_dataset2 as create_dataset_thor

tests/st/networks/test_gpu_lenet.py

@@ -27,12 +27,11 @@ from mindspore import Tensor, ParameterTuple
 from mindspore.common import dtype as mstype
 from mindspore.dataset.vision import Inter
 from mindspore.nn import Dense, TrainOneStepCell, WithLossCell, ForwardValueAndGrad
-from mindspore.train.metrics import Accuracy
+from mindspore.train import Accuracy
 from mindspore.nn.optim import Momentum
 from mindspore.ops import operations as P
 from mindspore.ops import functional as F
-from mindspore.train import Model
-from mindspore.train.callback import LossMonitor
+from mindspore.train import Model, LossMonitor
 from mindspore.common.initializer import TruncatedNormal
 
 context.set_context(mode=context.GRAPH_MODE, device_target="GPU")

tests/st/ops/ascend/test_shape.py

@@ -18,7 +18,8 @@ import mindspore.context as context
 import mindspore.nn as nn
 import mindspore.dataset as ds
 from mindspore.ops import operations as P
-from mindspore import Model, Tensor
+from mindspore import Tensor
+from mindspore.train import Model
 import mindspore as ms
 
 context.set_context(mode=context.GRAPH_MODE,

tests/st/ops/ascend/test_tbe_ops/test_batchmatmul.py

@@ -18,7 +18,7 @@ from mindspore import context
 from mindspore.common.tensor import Tensor
 from mindspore.nn import Cell
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_fast_gelu.py

@@ -18,7 +18,7 @@ import pytest
 from mindspore import context
 from mindspore.common.tensor import Tensor
 from mindspore.nn import FastGelu
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_gelu.py

@@ -19,7 +19,7 @@ from mindspore import context
 from mindspore import log as logger
 from mindspore.common.tensor import Tensor
 from mindspore.nn import GELU
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_greater.py

@@ -20,7 +20,7 @@ from mindspore import log as logger
 from mindspore.common.tensor import Tensor
 from mindspore.nn import Cell
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_layernorm.py

@@ -20,7 +20,7 @@ from mindspore import log as logger
 from mindspore.common.tensor import Tensor
 from mindspore.nn import Cell
 from mindspore.nn import LayerNorm
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_maximum.py

@@ -19,7 +19,7 @@ import mindspore.context as context
 import mindspore.nn as nn
 from mindspore import Tensor
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_minimum.py

@@ -19,7 +19,7 @@ import mindspore.context as context
 import mindspore.nn as nn
 from mindspore import Tensor
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_pow.py

@@ -19,7 +19,7 @@ import mindspore.context as context
 from mindspore import Tensor
 from mindspore.nn import Cell
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_select.py

@@ -21,7 +21,7 @@ from mindspore import Tensor
 from mindspore.nn import Cell
 import mindspore.ops as ops
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_stridedslice.py

@@ -18,7 +18,7 @@ import mindspore.context as context
 import mindspore.ops.operations as P
 from mindspore.common.tensor import Tensor
 from mindspore.nn import Cell
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_tanh.py

@@ -18,7 +18,7 @@ import mindspore.context as context
 import mindspore.nn as nn
 from mindspore import Tensor
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_tanh_grad.py

@@ -18,7 +18,7 @@ import mindspore.context as context
 import mindspore.nn as nn
 from mindspore import Tensor
 from mindspore.ops.operations import _grad_ops as G
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(device_target="Ascend")
 

tests/st/ops/ascend/test_tbe_ops/test_transpose_d.py

@@ -18,7 +18,7 @@ import mindspore.context as context
 import mindspore.nn as nn
 from mindspore import Tensor
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 context.set_context(device_target="Ascend")
 

tests/st/place/general_split/test_general_split_lenet.py

@@ -15,9 +15,8 @@
 
 import mindspore.context as context
 from mindspore import set_seed
-from mindspore.train.metrics import Accuracy
-from mindspore.train import Model
-from mindspore.train.callback import LossMonitor, TimeMonitor
+from mindspore.train import Accuracy
+from mindspore.train import Model, LossMonitor, TimeMonitor
 from mindspore.communication.management import init
 
 from src.lenet import Net, get_optimizer, get_loss, get_dataset

tests/st/place/optimizer_split/test_optimizer_split.py

@@ -15,9 +15,9 @@
 
 import mindspore.context as context
 from mindspore import set_seed
-from mindspore.train.metrics import Accuracy
+from mindspore.train import Accuracy
 from mindspore.train import Model
-from mindspore.train.callback import LossMonitor, TimeMonitor
+from mindspore.train import LossMonitor, TimeMonitor
 from mindspore.communication.management import init, get_rank
 
 from src.net import SplitRefWithoutOptimNet, SplitOptimNet, get_optimizer, get_loss, get_dataset

tests/st/place/ref_split/test_ref_split.py

@@ -15,7 +15,7 @@
 
 import mindspore.context as context
 from mindspore import set_seed
-from mindspore.train.metrics import Accuracy
+from mindspore.train import Accuracy
 from mindspore.train import Model
 from mindspore.communication.management import init, get_rank
 

tests/st/place/simple_dynamic_shape/test_simple_dynamic_shape.py

@@ -15,9 +15,8 @@
 
 import mindspore.context as context
 from mindspore import set_seed
-from mindspore.train.metrics import Accuracy
-from mindspore.train import Model
-from mindspore.train.callback import TimeMonitor
+from mindspore.train import Accuracy
+from mindspore.train import Model, TimeMonitor
 from mindspore.communication.management import init
 
 from src.dynamic_embedding import Net, get_optimizer, get_loss, get_dataset

tests/st/probability/zhusuan/vae/vae_mnist.py

@@ -23,8 +23,7 @@ import mindspore.nn as nn
 
 from mindspore import context
 from mindspore import Tensor
-from mindspore.train import Model
-from mindspore.train.callback import LossMonitor
+from mindspore.train import Model, LossMonitor
 from mindspore.ops import operations as P
 from mindspore.common import dtype as mstype
 

tests/st/profiler/run_net.py

@@ -17,7 +17,7 @@ from argparse import ArgumentParser
 
 from mindspore import dataset as ds
 from mindspore import nn, Tensor, context
-from mindspore.train.metrics import Accuracy
+from mindspore.train import Accuracy
 from mindspore.nn.optim import Momentum
 from mindspore.dataset.transforms import transforms as C
 from mindspore.dataset.vision import transforms as CV

tests/st/profiler/test_profiler.py

@@ -22,14 +22,13 @@ import pytest
 
 from mindspore import dataset as ds
 from mindspore import nn, Tensor, context
-from mindspore.train.metrics import Accuracy
 from mindspore.nn.optim import Momentum
 from mindspore.dataset.transforms import transforms as C
 from mindspore.dataset.vision import transforms as CV
 from mindspore.dataset.vision import Inter
 from mindspore.common import dtype as mstype
 from mindspore.common.initializer import TruncatedNormal
-from mindspore.train import Model
+from mindspore.train import Model, Accuracy
 from mindspore import Profiler
 
 

tests/st/ps/embedding_cache/src/model.py

@@ -20,9 +20,8 @@ import mindspore
 from mindspore.nn import Cell, Flatten, Dense
 from mindspore.nn import EmbeddingLookup, SoftmaxCrossEntropyWithLogits
 from mindspore.nn import Adam
-from mindspore.train import Model
-from mindspore.train.callback import CheckpointConfig, ModelCheckpoint
-from mindspore.train.metrics import Accuracy
+from mindspore.train import Model, CheckpointConfig, ModelCheckpoint
+from mindspore.train import Accuracy
 from mindspore.common import set_seed
 from mindspore.communication.management import get_rank
 import mindspore.ops.operations as op

tests/st/ps/full_ps/test_full_ps_lenet.py

@@ -23,9 +23,7 @@ import mindspore.dataset.vision as CV
 import mindspore.nn as nn
 from mindspore.common import dtype as mstype
 from mindspore.dataset.vision import Inter
-from mindspore.train.metrics import Accuracy
-from mindspore.train import Model
-from mindspore.train.callback import LossMonitor
+from mindspore.train import Model, LossMonitor, Accuracy
 from mindspore.common.initializer import TruncatedNormal
 from mindspore.communication.management import init
 

tests/st/pynative/test_pynative_layernorm_input_and_argmaxwithvalue.py

@@ -21,7 +21,7 @@ from mindspore import Tensor, context
 from mindspore.nn import LayerNorm, Cell
 from mindspore.common import ParameterTuple
 from mindspore.ops.composite import GradOperation
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 class _Grad(Cell):
     def __init__(self, grad, network, wrt_params=False, real_inputs_count=None):

tests/st/pynative/test_pynative_resnet50_ascend.py

@@ -35,10 +35,10 @@ from mindspore.nn import Cell
 from mindspore.ops import operations as P
 from mindspore.ops import composite as CP
 from mindspore.nn.optim.momentum import Momentum
-from mindspore.train.callback import Callback
+from mindspore.train import Callback
 from mindspore.nn.loss import SoftmaxCrossEntropyWithLogits
 from mindspore.train.loss_scale_manager import FixedLossScaleManager
-from mindspore.train.model import Model
+from mindspore.train import Model
 from mindspore.context import ParallelMode
 import mindspore.communication.management as D
 MINDSPORE_HCCL_CONFIG_PATH = "/home/workspace/mindspore_config/hccl/rank_table_8p.json"

tests/st/summary/test_summary_collector.py

@@ -25,12 +25,11 @@ import pytest
 from mindspore.common import set_seed
 from mindspore import nn, Tensor, context
 from mindspore.common.initializer import Normal
-from mindspore.train.metrics import Loss
+from mindspore.train import Loss
 from mindspore.nn.optim import Momentum
 from mindspore.ops import operations as P
 from mindspore.train import Model
-from mindspore.train.callback import SummaryCollector, SummaryLandscape
-from mindspore.train.summary.summary_record import SummaryRecord
+from mindspore import SummaryCollector, SummaryLandscape, SummaryRecord
 from tests.st.summary.dataset import create_mnist_dataset
 from tests.summary_utils import SummaryReader
 from tests.security_utils import security_off_wrap

tests/st/summary/test_summary_ops.py

@@ -22,7 +22,7 @@ import pytest
 
 from mindspore import nn, Tensor, context
 from mindspore.common.initializer import Normal
-from mindspore.train.metrics import Loss
+from mindspore.train import Loss
 from mindspore.nn.optim import Momentum
 from mindspore.ops import operations as P
 from mindspore.train import Model

tests/st/tbe_networks/resnet_cifar.py

@@ -20,7 +20,7 @@ from resnet import resnet50
 
 import mindspore.common.dtype as mstype
 import mindspore.ops.functional as F
-from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor
+from mindspore.train import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor
 from mindspore.train.serialization import load_checkpoint, load_param_into_net
 import mindspore.dataset as ds
 import mindspore.dataset.transforms as C
@@ -31,7 +31,7 @@ from mindspore import context
 from mindspore.communication.management import init
 from mindspore.nn.optim.momentum import Momentum
 from mindspore.ops import operations as P
-from mindspore.train.model import Model
+from mindspore.train import Model
 from mindspore.context import ParallelMode
 
 random.seed(1)

tests/st/tbe_networks/test_resnet_cifar_1p.py

@@ -28,8 +28,8 @@ from mindspore import Tensor
 from mindspore import context
 from mindspore.nn.optim.momentum import Momentum
 from mindspore.ops import operations as P
-from mindspore.train.callback import Callback
-from mindspore.train.model import Model
+from mindspore.train import Callback
+from mindspore.train import Model
 
 random.seed(1)
 np.random.seed(1)

tests/st/tbe_networks/test_resnet_cifar_8p.py

@@ -30,8 +30,7 @@ from mindspore import context
 from mindspore.communication.management import init
 from mindspore.nn.optim.momentum import Momentum
 from mindspore.ops import operations as P
-from mindspore.train.callback import Callback
-from mindspore.train.model import Model
+from mindspore.train import Callback, Model
 from mindspore.context import ParallelMode
 
 random.seed(1)

tests/st/train/test_early_stopping_and_reduce_LR_on_plateau_callbacks.py

@@ -23,8 +23,8 @@ from mindspore import nn, Model
 from mindspore import dataset as ds
 from mindspore.nn.optim import Momentum
 from mindspore.common.initializer import Normal
-from mindspore.train.callback import RunContext, _InternalCallbackParam, \
-    _CallbackManager, ReduceLROnPlateau, EarlyStopping
+from mindspore.train import RunContext, ReduceLROnPlateau, EarlyStopping
+from mindspore.train.callback import _InternalCallbackParam, _CallbackManager
 
 
 def get_data(num, w=4.0, b=5.0):

tests/st/train/test_fit.py

@@ -17,9 +17,9 @@
 
 import pytest
 import numpy as np
-from mindspore import Model, nn, Tensor
+from mindspore import nn, Tensor
 from mindspore.common.initializer import Normal
-from mindspore.train.callback import Callback, TimeMonitor, LossMonitor
+from mindspore.train import Callback, TimeMonitor, LossMonitor, Model
 from mindspore import dataset as ds
 
 

tests/st/train/test_on_request_exit_callback.py

@@ -23,11 +23,10 @@ from multiprocessing import Process
 import numpy as np
 import pytest
 
-from mindspore import Model
 from mindspore import nn, context
 from mindspore import dataset as ds
 from mindspore.common.initializer import TruncatedNormal
-from mindspore.train.callback import Callback, OnRequestExit, LossMonitor
+from mindspore.train import Callback, OnRequestExit, LossMonitor, Model
 
 
 def conv(in_channels, out_channels, kernel_size, stride=1, padding=0):

tests/ut/cpp/python_input/gtest_input/pipeline/parse/parse_compile.py

@@ -24,7 +24,7 @@ import numpy as np
 import mindspore.nn as nn
 from mindspore.common.tensor import Tensor
 from mindspore.nn.optim import Momentum
-from mindspore.train.model import Model
+from mindspore.train import Model
 
 log = logging.getLogger("test")
 log.setLevel(level=logging.ERROR)

tests/ut/cpp/python_input/gtest_input/pipeline/parse/parser_integrate.py

@@ -23,7 +23,7 @@ from mindspore.common.api import jit, _cell_graph_executor
 from mindspore.common.parameter import Parameter
 from mindspore.common.tensor import Tensor
 from mindspore.ops import functional as F
-from mindspore.train.model import Model
+from mindspore.train import Model
 from tests.ut.python.model.resnet import resnet50
 
 

tests/ut/python/dataset/test_callbacks.py

@@ -20,8 +20,7 @@ import pytest
 from mindspore import context
 from mindspore import log as logger
 from mindspore.dataset.callback import DSCallback, WaitedDSCallback
-from mindspore.train import Model
-from mindspore.train.callback import Callback
+from mindspore.train import Model, Callback
 
 import mindspore.dataset as ds
 import mindspore.nn as nn

tests/ut/python/metrics/test_accuracy.py

@@ -18,7 +18,7 @@ import numpy as np
 import pytest
 
 from mindspore import Tensor
-from mindspore.train.metrics import Accuracy
+from mindspore.train import Accuracy
 
 
 def test_classification_accuracy():

tests/ut/python/metrics/test_auc.py

@@ -17,7 +17,7 @@
 import math
 import numpy as np
 from mindspore import Tensor
-from mindspore.train.metrics import ROC, auc
+from mindspore.train import ROC, auc
 
 
 def test_auc():

tests/ut/python/metrics/test_bleu_score.py

@@ -15,7 +15,7 @@
 """test_bleu_score"""
 import math
 import pytest
-from mindspore.train.metrics import BleuScore
+from mindspore.train import BleuScore
 
 
 def test_bleu_score():

tests/ut/python/metrics/test_confusion_matrix.py

@@ -16,7 +16,7 @@
 import numpy as np
 import pytest
 from mindspore import Tensor
-from mindspore.train.metrics import ConfusionMatrix
+from mindspore.train import ConfusionMatrix
 
 
 def test_confusion_matrix():

tests/ut/python/metrics/test_confusion_matrix_metric.py

@@ -16,7 +16,7 @@
 import numpy as np
 import pytest
 from mindspore import Tensor
-from mindspore.train.metrics import ConfusionMatrixMetric
+from mindspore.train import ConfusionMatrixMetric
 
 
 def test_confusion_matrix_metric():

tests/ut/python/metrics/test_cosine_similarity.py

@@ -16,7 +16,7 @@
 import pytest
 import numpy as np
 from sklearn.metrics import pairwise
-from mindspore.train.metrics import CosineSimilarity
+from mindspore.train import CosineSimilarity
 
 
 def test_cosine_similarity():

tests/ut/python/metrics/test_dice.py

@@ -17,7 +17,7 @@ import math
 import numpy as np
 import pytest
 from mindspore import Tensor
-from mindspore.train.metrics import get_metric_fn, Dice
+from mindspore.train import get_metric_fn, Dice
 
 
 def test_classification_dice():

tests/ut/python/metrics/test_error.py

@@ -18,7 +18,7 @@ import numpy as np
 import pytest
 
 from mindspore import Tensor
-from mindspore.train.metrics import MAE, MSE
+from mindspore.train import MAE, MSE
 
 
 def test_MAE():

tests/ut/python/metrics/test_fbeta.py

@@ -17,7 +17,7 @@ import numpy as np
 import pytest
 
 from mindspore import Tensor
-from mindspore.train.metrics import get_metric_fn, Fbeta
+from mindspore.train import get_metric_fn, Fbeta
 
 
 def test_classification_fbeta():

tests/ut/python/metrics/test_hausdorff_distance.py

@@ -18,7 +18,7 @@ import math
 import numpy as np
 import pytest
 from mindspore import Tensor
-from mindspore.train.metrics import get_metric_fn, HausdorffDistance
+from mindspore.train import get_metric_fn, HausdorffDistance
 
 
 def test_hausdorff_distance():

tests/ut/python/metrics/test_loss.py

@@ -17,7 +17,7 @@ import numpy as np
 import pytest
 
 from mindspore import Tensor
-from mindspore.train.metrics import Loss
+from mindspore.train import Loss
 
 
 def test_loss_inputs_error():

tests/ut/python/metrics/test_mean_surface_distance.py

@@ -18,7 +18,7 @@ import math
 import numpy as np
 import pytest
 from mindspore import Tensor
-from mindspore.train.metrics import get_metric_fn, MeanSurfaceDistance
+from mindspore.train import get_metric_fn, MeanSurfaceDistance
 
 
 def test_mean_surface_distance():

tests/ut/python/metrics/test_metric_factory.py

@@ -17,7 +17,7 @@ import math
 import numpy as np
 
 from mindspore import Tensor
-from mindspore.train.metrics import get_metric_fn, rearrange_inputs
+from mindspore.train import get_metric_fn, rearrange_inputs
 
 
 def test_classification_accuracy():

tests/ut/python/metrics/test_occlusion_sensitivity.py

@@ -17,7 +17,7 @@ import pytest
 import numpy as np
 from mindspore import nn, context
 from mindspore.common.tensor import Tensor
-from mindspore.train.metrics import OcclusionSensitivity
+from mindspore.train import OcclusionSensitivity
 context.set_context(mode=context.GRAPH_MODE)
 
 class DenseNet(nn.Cell):

tests/ut/python/metrics/test_perplexity.py

@@ -18,7 +18,7 @@ import math
 import numpy as np
 import pytest
 from mindspore import Tensor
-from mindspore.train.metrics import get_metric_fn, Perplexity
+from mindspore.train import get_metric_fn, Perplexity
 
 
 def test_perplexity():

tests/ut/python/metrics/test_precision.py

@@ -18,7 +18,7 @@ import numpy as np
 import pytest
 
 from mindspore import Tensor
-from mindspore.train.metrics import Precision
+from mindspore.train import Precision
 
 
 def test_classification_precision():

tests/ut/python/metrics/test_recall.py

@@ -18,7 +18,7 @@ import numpy as np
 import pytest
 
 from mindspore import Tensor
-from mindspore.train.metrics import Recall
+from mindspore.train import Recall
 
 
 def test_classification_recall():

tests/ut/python/metrics/test_roc.py

@@ -17,7 +17,7 @@
 import numpy as np
 import pytest
 from mindspore import Tensor
-from mindspore.train.metrics import ROC
+from mindspore.train import ROC
 
 
 def test_roc():

tests/ut/python/metrics/test_root_mean_square_distance.py

@@ -18,7 +18,7 @@ import math
 import numpy as np
 import pytest
 from mindspore import Tensor
-from mindspore.train.metrics import get_metric_fn, RootMeanSquareDistance
+from mindspore.train import get_metric_fn, RootMeanSquareDistance
 
 
 def test_root_mean_square_distance():

tests/ut/python/metrics/test_topk.py

@@ -18,7 +18,7 @@ import numpy as np
 import pytest
 
 from mindspore import Tensor
-from mindspore.train.metrics import TopKCategoricalAccuracy, Top1CategoricalAccuracy, Top5CategoricalAccuracy
+from mindspore.train import TopKCategoricalAccuracy, Top1CategoricalAccuracy, Top5CategoricalAccuracy
 
 
 def test_type_topk():

tests/ut/python/parallel/test_auto_parallel_pangu_alpha_shard_propagation.py

@@ -27,7 +27,7 @@ from mindspore import context
 from mindspore.nn.wrap.cell_wrapper import _VirtualDatasetCell
 from mindspore.parallel import set_algo_parameters
 from mindspore.nn.layer.activation import get_activation
-from mindspore.train.model import Model
+from mindspore.train import Model
 from mindspore.common.api import _cell_graph_executor
 from tests.dataset_mock import MindData
 context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")

tests/ut/python/parallel/test_auto_parallel_resnet.py

@@ -29,7 +29,7 @@ from mindspore.ops import operations as P
 from mindspore.parallel import _cost_model_context as cost_model_context
 from mindspore.parallel import set_algo_parameters
 from mindspore.parallel._utils import _reset_op_id as resset_op_id
-from mindspore.train.model import Model
+from mindspore.train import Model
 from mindspore.context import ParallelMode
 from mindspore.communication._comm_helper import GlobalComm