forked from mindspore-Ecosystem/mindspore
!449 Fix the error in example of mindspore.model
Merge pull request !449 from Simson/push-to-opensource
This commit is contained in:
commit
24fff5fe63
|
@ -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