Browse Source

!6816 modify nasnet parallelmode

Merge pull request !6816 from panfengfeng/fix_nasnet_scripts
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
fe934520e6
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      model_zoo/official/cv/nasnet/src/nasnet_a_mobile.py
  2. +1
    -1
      model_zoo/official/cv/nasnet/train.py

+ 1
- 1
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


+ 1
- 1
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


Loading…
Cancel
Save