forked from mindspore-Ecosystem/mindspore
fix the error in example of mindspore.model
This commit is contained in:
parent
10877f077e
commit
8a2a1eb310
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue