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