diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 8a67a1305718e36744988da0840372d105c1572f..be3939d450120ab2c5fe8bc675e7caa7e01907e3 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -71,7 +71,7 @@ class Model: >>> self.bn = nn.BatchNorm2d(64) >>> self.relu = nn.ReLU() >>> self.flatten = nn.Flatten() - >>> self.fc = nn.Dense(64*222*222, 3) # padding=0 + >>> self.fc = nn.Dense(64*224*224, 12) # padding=0 >>> >>> def construct(self, x): >>> x = self.conv(x)