conflicts
This commit is contained in:
commit
9a3aa18c17
29
CITATION.bib
Normal file
29
CITATION.bib
Normal file
@ -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)
|
Dense(288, 10), softmax)
|
||||||
|
|
||||||
@test size(m(r)) == (10, 5)
|
@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
|
end
|
||||||
|
Loading…
Reference in New Issue
Block a user