d360438 · grenade · git.iko.soy

git.iko.soy/grenade

Update hmatrix

commitd360438fc0bdae072292325dddd972c05e6ebb40 authorHuw Campbell <huw.campbell@ambiata.com> dateWed 7 Dec 2016, 14:50 (10 years ago) parent2cb93c6

3 files changed+3−4

unifiedsplit

grenade.cabal+1−1

@@ -76,7 +76,7 @@ executable mnist

7676 , optparse-applicative == 0.12.*

7777 , text == 1.2.*

7878 , mtl >= 2.2.1 && < 2.3

79− , hmatrix

79+ , hmatrix >= 0.18 && < 0.19

8080 , transformers

8181 , singletons

8282 , MonadRandom

lib/hmatrixdeleted−1

Submodule changed.

src/Grenade/Core/Runner.hs+2−2

@@ -16,7 +16,7 @@ import Data.Singletons.Prelude

1616import Grenade.Core.Network

1717import Grenade.Core.Shape

1818

19−-- | Drive and network and collect it's back propogated gradients.

19+-- | Drive and network and collect its back propogated gradients.

2020backPropagate :: forall input output shapes layers. (Head shapes ~ input, Last shapes ~ output)

2121 => Network layers shapes -> S' input -> S' output -> Gradients layers

2222backPropagate network input target =

@@ -29,7 +29,7 @@ backPropagate network input target =

2929 -- handle input from the beginning, feeding upwards.

3030 go !x (layer :~> n)

3131 = let y = runForwards layer x

32− -- recursively run the rest of the network, and get the layer from above.

32+ -- recursively run the rest of the network, and get the gradients from above.

3333 (n', dWs') = go y n

3434 -- calculate the gradient for this layer to pass down,

3535 (layer', dWs) = runBackards layer x dWs'