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.ops.composite as C
import mindspore.common.dtype as mstype import mindspore.common.dtype as mstype
from mindspore.nn.wrap.grad_reducer import DistributedGradReducer 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 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 Tensor
from mindspore import context 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.communication.management import init, get_rank, get_group_size
from mindspore.nn.optim.rmsprop import RMSProp from mindspore.nn.optim.rmsprop import RMSProp
from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor


Loading…
Cancel
Save