Browse Source

fix the error in example of mindspore.model

tags/v0.2.0-alpha
simson 6 years ago
parent
commit
8a2a1eb310
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/model.py

+ 1
- 1
mindspore/train/model.py View File

@@ -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…
Cancel
Save