|
|
|
@ -28,6 +28,7 @@ from mindspore.nn.optim.momentum import Momentum
|
|
|
|
|
from mindspore.train.model import Model, ParallelMode
|
|
|
|
|
from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor
|
|
|
|
|
from mindspore.train.loss_scale_manager import FixedLossScaleManager
|
|
|
|
|
from mindspore.train.serialization import load_checkpoint, load_param_into_net
|
|
|
|
|
import mindspore.dataset.engine as de
|
|
|
|
|
from mindspore.communication.management import init
|
|
|
|
|
import mindspore.nn as nn
|
|
|
|
|