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

removed extra expand and fixed bug

Browse Source
This commit is contained in:
tejank10 2018-04-16 01:18:26 +05:30
parent 2f5473d435
commit 2ef25775c6
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/layers/conv.jl
View File

@ -26,12 +26,12 @@ end
Conv(w::AbstractArray{T}, b::AbstractVector{T}, σ = identity;
stride = 1, pad = 0) where T =
Conv(σ, w, b, expand(Val{ndims(w) - 2}, stride),
pad = expand(Val{ndims(w) - 2}, pad))
expand(Val{ndims(w) - 2}, pad))
Conv(k::NTuple{N,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity; init = initn,
stride = 1, pad = 0) where N =
Conv(param(init(k..., ch...)), param(zeros(ch[2])), σ,
stride = expand(Val{N}, stride), pad = expand(Val{N}, pad))
Conv(param(init(k..., ch...)), param(zeros(ch[2])), σ;
stride = stride, pad = pad)
Flux.treelike(Conv)