diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 46e4f421f7d34479577776206e6bdba1a50ba122..a9f59207bbf6114bdeabeb15362bdca165b0f097 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)