mirror of
https://github.com/HuwCampbell/grenade.git
synced 2024-11-21 13:22:32 +03:00
fix typo
This commit is contained in:
parent
766b962341
commit
83cb4e4ec2
@ -53,7 +53,7 @@ instance (KnownNat i, KnownNat o) => Layer (FullyConnected i o) ('D1 i) ('D1 o)
|
||||
runBackwards (FullyConnected (FullyConnected' _ wN) _) x (S1D dEdy) =
|
||||
let wB' = dEdy
|
||||
mm' = dEdy `outer` x
|
||||
-- calcluate derivatives for next step
|
||||
-- calculate derivatives for next step
|
||||
dWs = tr wN #> dEdy
|
||||
in (FullyConnected' wB' mm', S1D dWs)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user