diff --git a/mindspore/nn/layer/container.py b/mindspore/nn/layer/container.py index f588100e80..f084cdc9e2 100644 --- a/mindspore/nn/layer/container.py +++ b/mindspore/nn/layer/container.py @@ -259,8 +259,13 @@ class CellList(_CellListBase, Cell): >>> cell_ls.append(relu) >>> cell_ls CellList< - (0): Conv2d - (1): BatchNorm2d + (0): Conv2d + (1): BatchNorm2d (2): ReLU<> > """ diff --git a/mindspore/nn/loss/loss.py b/mindspore/nn/loss/loss.py index 652dfcc926..21c1c68e74 100644 --- a/mindspore/nn/loss/loss.py +++ b/mindspore/nn/loss/loss.py @@ -512,7 +512,7 @@ class MultiClassDiceLoss(_Loss): >>> y = Tensor(np.array([[0, 1], [1, 0], [0, 1]]), mstype.float32) >>> output = loss(y_pred, y) >>> print(output) - [0.3283009] + 0.3283009 """ def __init__(self, weights=None, ignore_indiex=None, activation="softmax"): super(MultiClassDiceLoss, self).__init__()