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

Added Float64 tests for DepthwiseConv

Browse Source
This commit is contained in:
Manjunath Bhat 2019-03-09 20:17:22 +05:30 committed by GitHub
parent 704be49483
commit d4a1d33a31
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
test/layers/conv.jl
View File

@ -32,4 +32,14 @@ end
m2 = DepthwiseConv((2, 2), 3) m2 = DepthwiseConv((2, 2), 3)
@test size(m2(r), 3) == 3 @test size(m2(r), 3) == 3
x = zeros(Float64, 28, 28, 3, 5)
m3 = DepthwiseConv((2, 2), 3 => 5)
@test size(m3(r), 3) == 15
m4 = DepthwiseConv((2, 2), 3)
@test size(m4(r), 3) == 3
end end