@@ -119,15 +119,7 @@ randomConvolution = do
119119 mm = konst 0
120120 return $ Convolution wN mm
121121
122−instance ( Monad m
123− , KnownNat kernelRows
124− , KnownNat kernelCols
125− , KnownNat channels
126− , KnownNat filters
127− , KnownNat strideRows
128− , KnownNat strideCols
129− , kernelFlattened ~ (kernelRows * kernelColumns * channels)
130− ) => UpdateLayer m (Convolution channels filters kernelRows kernelCols strideRows strideCols) where
122+instance ( Monad m ) => UpdateLayer m (Convolution channels filters kernelRows kernelCols strideRows strideCols) where
131123 type Gradient (Convolution channels filters kernelRows kernelCols strideRows strideCols) = (Convolution' channels filters kernelRows kernelCols strideRows strideCols)
132124 runUpdate LearningParameters {..} (Convolution oldKernel oldMomentum) (Convolution' kernelGradient) = do
133125 let newMomentum = konst learningMomentum * oldMomentum - konst learningRate * kernelGradient