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

fix typo in conv.jl (fixes #133)

Browse Source
This commit is contained in:
Mehul Tikekar 2018-01-08 16:45:06 -05:00
parent 468f641f66
commit 2fef799109
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/conv.jl
View File

@ -19,7 +19,7 @@ end
Conv2D(k::NTuple{2,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity; Conv2D(k::NTuple{2,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity;
init = initn, stride = 1, pad = 0) = init = initn, stride = 1, pad = 0) =
Conv2D(σ, param(initn(k..., ch...)), stride, pad) Conv2D(σ, param(init(k..., ch...)), stride, pad)
Flux.treelike(Conv2D) Flux.treelike(Conv2D)