File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -58,7 +58,7 @@ You can extract the first or last layer of a composed model using `getFirstLayer
58
58
59
59
``` haskell
60
60
-- Assume 'model' is an instance of `MLP`
61
- ( MLP ( model :: a )) <- sample mlpSpec
61
+ model <- sample mlpSpec
62
62
let firstLayer = getFirstLayer model -- Get the first Linear layer
63
63
let lastLayer = getLastLayer model -- Get the LogSoftmax layer
64
64
let modelWithoutLast = dropLastLayer model
@@ -82,7 +82,7 @@ The `toOutputShapes` function allows you to get the shapes of each layer's outpu
82
82
83
83
``` haskell
84
84
-- Assume 'model' is an instance of `MLP`
85
- ( MLP model) <- sample mlpSpec
85
+ model <- sample mlpSpec
86
86
let input = ones' [2 ,784 ]
87
87
let outputShapes = toOutputShapes model input
88
88
-- outputShapes will be a HList containing the shape of each layer's output.
You can’t perform that action at this time.
0 commit comments