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
5858
5959``` haskell
6060-- Assume 'model' is an instance of `MLP`
61- ( MLP ( model :: a )) <- sample mlpSpec
61+ model <- sample mlpSpec
6262let firstLayer = getFirstLayer model -- Get the first Linear layer
6363let lastLayer = getLastLayer model -- Get the LogSoftmax layer
6464let 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
8686let input = ones' [2 ,784 ]
8787let outputShapes = toOutputShapes model input
8888-- 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