conflicts
This commit is contained in:
commit
9a3aa18c17
|
@ -0,0 +1,29 @@
|
|||
@article{Flux.jl-2018,
|
||||
author = {Michael Innes and
|
||||
Elliot Saba and
|
||||
Keno Fischer and
|
||||
Dhairya Gandhi and
|
||||
Marco Concetto Rudilosso and
|
||||
Neethu Mariya Joy and
|
||||
Tejan Karmali and
|
||||
Avik Pal and
|
||||
Viral Shah},
|
||||
title = {Fashionable Modelling with Flux},
|
||||
journal = {CoRR},
|
||||
volume = {abs/1811.01457},
|
||||
year = {2018},
|
||||
url = {http://arxiv.org/abs/1811.01457},
|
||||
archivePrefix = {arXiv},
|
||||
eprint = {1811.01457},
|
||||
timestamp = {Thu, 22 Nov 2018 17:58:30 +0100},
|
||||
biburl = {https://dblp.org/rec/bib/journals/corr/abs-1811-01457},
|
||||
bibsource = {dblp computer science bibliography, https://dblp.org}
|
||||
}
|
||||
|
||||
@article{innes:2018,
|
||||
author = {Mike Innes},
|
||||
title = {Flux: Elegant Machine Learning with Julia},
|
||||
journal = {Journal of Open Source Software},
|
||||
year = {2018},
|
||||
doi = {10.21105/joss.00602},
|
||||
}
|
|
@ -80,5 +80,30 @@ end
|
|||
Dense(288, 10), softmax)
|
||||
|
||||
@test size(m(r)) == (10, 5)
|
||||
|
||||
end
|
||||
|
||||
@testset "Conv with non quadratic window #700" begin
|
||||
data = zeros(Float32, 7,7,1,1)
|
||||
data[4,4,1,1] = 1
|
||||
|
||||
l = Conv((3,3), 1=>1)
|
||||
expected = zeros(eltype(l.weight),5,5,1,1)
|
||||
expected[2:end-1,2:end-1,1,1] = l.weight
|
||||
@test expected == l(data)
|
||||
|
||||
l = Conv((3,1), 1=>1)
|
||||
expected = zeros(eltype(l.weight),5,7,1,1)
|
||||
expected[2:end-1,4,1,1] = l.weight
|
||||
@test expected == l(data)
|
||||
|
||||
l = Conv((1,3), 1=>1)
|
||||
expected = zeros(eltype(l.weight),7,5,1,1)
|
||||
expected[4,2:end-1,1,1] = l.weight
|
||||
@test expected == l(data)
|
||||
|
||||
@test begin
|
||||
# we test that the next expression does not throw
|
||||
randn(Float32, 10,10,1,1) |> Conv((6,1), 1=>1, Flux.σ)
|
||||
true
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in New Issue