|
|
|
@ -18,7 +18,7 @@ import os
|
|
|
|
|
|
|
|
|
|
from mindspore import Tensor
|
|
|
|
|
from mindspore import context
|
|
|
|
|
from mindspore import ParallelMode
|
|
|
|
|
from mindspore.context import ParallelMode
|
|
|
|
|
from mindspore.communication.management import init, get_rank, get_group_size
|
|
|
|
|
from mindspore.nn.optim.rmsprop import RMSProp
|
|
|
|
|
from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor
|
|
|
|
|