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

gpu fix

Browse Source
This commit is contained in:
Mike J Innes 2018-02-17 12:41:53 +00:00
parent 11511982a4
commit 989adcdc7d
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

@ -29,7 +29,7 @@ Conv2D(k::NTuple{2,Integer}, ch::Pair{<:Integer,<:Integer}, σ = identity;
Flux.treelike(Conv2D) Flux.treelike(Conv2D)
function (c::Conv2D)(x) function (c::Conv2D)(x)
σ, b = c.σ, reshape(c.bias, 1, 1, :) σ, b = c.σ, reshape(c.bias, 1, 1, :, 1)
σ.(conv2d(x, c.weight, stride = c.stride, padding = c.pad) .+ b) σ.(conv2d(x, c.weight, stride = c.stride, padding = c.pad) .+ b)
end end