HuwCampbell/grenade
1
1
Fork 0
mirror of https://github.com/HuwCampbell/grenade.git synced 2024-11-22 06:55:13 +03:00
Code Issues Projects Releases Wiki Activity

fix typo

Browse Source
This commit is contained in:
sorki 2023-12-08 20:05:06 +01:00 committed by Erik de Castro Lopo
parent 766b962341
commit 83cb4e4ec2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Grenade/Layers/FullyConnected.hs
View File

@ -53,7 +53,7 @@ instance (KnownNat i, KnownNat o) => Layer (FullyConnected i o) ('D1 i) ('D1 o)
runBackwards (FullyConnected (FullyConnected' _ wN) _) x (S1D dEdy) = runBackwards (FullyConnected (FullyConnected' _ wN) _) x (S1D dEdy) =
let wB' = dEdy let wB' = dEdy
mm' = dEdy `outer` x mm' = dEdy `outer` x
-- calcluate derivatives for next step -- calculate derivatives for next step
dWs = tr wN #> dEdy dWs = tr wN #> dEdy
in (FullyConnected' wB' mm', S1D dWs) in (FullyConnected' wB' mm', S1D dWs)