TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Removed .data from LSTMCell

Browse Source
This commit is contained in:
thebhatman 2019-07-02 17:38:30 +05:30
parent 517219ba23
commit 3ee2a76f61
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/layers/recurrent.jl
View File

@ -109,7 +109,7 @@ function LSTMCell(in::Integer, out::Integer;
init = glorot_uniform) init = glorot_uniform)
cell = LSTMCell(init(out * 4, in), init(out * 4, out), init(out * 4), cell = LSTMCell(init(out * 4, in), init(out * 4, out), init(out * 4),
zeros(out), zeros(out)) zeros(out), zeros(out))
cell.b.data[gate(out, 2)] .= 1 cell.b[gate(out, 2)] .= 1
return cell return cell
end end