Skip to content

Commit 92c345f

Browse files
Update
1 parent 11ca756 commit 92c345f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ You can extract the first or last layer of a composed model using `getFirstLayer
5858

5959
```haskell
6060
-- Assume 'model' is an instance of `MLP`
61-
(MLP (model :: a)) <- sample mlpSpec
61+
model <- sample mlpSpec
6262
let firstLayer = getFirstLayer model -- Get the first Linear layer
6363
let lastLayer = getLastLayer model -- Get the LogSoftmax layer
6464
let modelWithoutLast = dropLastLayer model
@@ -82,7 +82,7 @@ The `toOutputShapes` function allows you to get the shapes of each layer's outpu
8282

8383
```haskell
8484
-- Assume 'model' is an instance of `MLP`
85-
(MLP model) <- sample mlpSpec
85+
model <- sample mlpSpec
8686
let input = ones' [2,784]
8787
let outputShapes = toOutputShapes model input
8888
-- outputShapes will be a HList containing the shape of each layer's output.

0 commit comments

Comments
 (0)