p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix nasnet scripts

This commit is contained in:
panfengfeng 2020-09-24 10:34:49 +08:00
parent 7f390467e9
commit 789296949b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/nasnet/src/nasnet_a_mobile.py
View File

@ -23,7 +23,7 @@ import mindspore.ops.functional as F
import mindspore.ops.composite as C
import mindspore.common.dtype as mstype
from mindspore.nn.wrap.grad_reducer import DistributedGradReducer
from mindspore.train.parallel_utils import ParallelMode
from mindspore.context import ParallelMode
from mindspore.parallel._utils import _get_device_num, _get_parallel_mode, _get_gradients_mean

2
model_zoo/official/cv/nasnet/train.py
View File

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