diff --git a/mindspore/nn/layer/basic.py b/mindspore/nn/layer/basic.py index f22dc3c6ab..5de8bf80e9 100644 --- a/mindspore/nn/layer/basic.py +++ b/mindspore/nn/layer/basic.py @@ -161,7 +161,7 @@ class Dropout(Cell): return out def extend_repr(self): - return 'keep_prob={}, dtype={}'.format(self.keep_prob, self.dtype) + return 'keep_prob={}'.format(self.keep_prob) class Flatten(Cell): diff --git a/mindspore/nn/layer/normalization.py b/mindspore/nn/layer/normalization.py index a19bdca97b..a7a2c7ddef 100644 --- a/mindspore/nn/layer/normalization.py +++ b/mindspore/nn/layer/normalization.py @@ -109,7 +109,8 @@ class _BatchNorm(Cell): epsilon=self.eps, momentum=self.momentum) self.bn_infer = P.BatchNorm(is_training=False, epsilon=self.eps, data_format=self.format) - self.enable_global_sync = self.is_global and (self.is_ge_backend or (self.is_graph_mode and self.is_ascend)) + self.enable_global_sync = self.is_global and (self.is_ge_backend or\ + (self.is_graph_mode and self._target == "Ascend")) data_parallel_strategy = ((1,), (1,)) data_parallel_strategy_one = ((1,), ())