Fix parallel mode import error

model_zoo/official/cv/deeplabv3/train.py

@@ -17,7 +17,8 @@
 import os
 import argparse
 from mindspore import context
-from mindspore.train.model import ParallelMode, Model
+from mindspore.train.model import Model
+from mindspore.context import ParallelMode
 import mindspore.nn as nn
 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig
 from mindspore.train.serialization import load_checkpoint, load_param_into_net

model_zoo/official/cv/mobilenetv2/src/utils.py

@@ -16,7 +16,7 @@
 from mindspore import context
 from mindspore import nn
 from mindspore.common import dtype as mstype
-from mindspore.train.model import ParallelMode
+from mindspore.context import ParallelMode
 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig
 from mindspore.communication.management import get_rank, init, get_group_size
 

model_zoo/official/cv/psenet/train.py

@@ -19,7 +19,8 @@ import mindspore.nn as nn
 from mindspore import context
 from mindspore.communication.management import init, get_rank
 from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, TimeMonitor
-from mindspore.train.model import Model, ParallelMode
+from mindspore.train.model import Model
+from mindspore.context import ParallelMode
 from mindspore.train.serialization import load_checkpoint, load_param_into_net
 from mindspore.common import set_seed
 

model_zoo/research/cv/ssd_ghostnet/train.py

@@ -22,7 +22,8 @@ import mindspore.nn as nn
 from mindspore import context, Tensor
 from mindspore.communication.management import init
 from mindspore.train.callback import CheckpointConfig, ModelCheckpoint, LossMonitor, TimeMonitor
-from mindspore.train import Model, ParallelMode
+from mindspore.train import Model
+from mindspore.context import ParallelMode
 from mindspore.train.serialization import load_checkpoint, load_param_into_net
 from src.ssd_ghostnet import SSD300, SSDWithLossCell, TrainingWrapper, ssd_ghostnet
 from src.config_ghostnet_13x import config