2019-09-10 14:02:43 +00:00
|
|
|
using Flux, Test, Random, Statistics, Documenter
|
2018-08-11 09:51:07 +00:00
|
|
|
using Random
|
2016-12-15 23:11:35 +00:00
|
|
|
|
2018-08-11 09:51:07 +00:00
|
|
|
Random.seed!(0)
|
2018-02-08 10:33:51 +00:00
|
|
|
|
2017-07-04 17:32:21 +00:00
|
|
|
@testset "Flux" begin
|
|
|
|
|
|
2018-09-05 14:39:00 +00:00
|
|
|
@info "Testing Basics"
|
|
|
|
|
|
2018-09-11 11:36:18 +00:00
|
|
|
include("utils.jl")
|
|
|
|
|
include("onehot.jl")
|
|
|
|
|
include("optimise.jl")
|
|
|
|
|
include("data.jl")
|
2018-09-05 14:39:00 +00:00
|
|
|
|
|
|
|
|
@info "Testing Layers"
|
|
|
|
|
|
2018-08-23 13:34:11 +00:00
|
|
|
include("layers/basic.jl")
|
2017-10-26 10:46:12 +00:00
|
|
|
include("layers/normalisation.jl")
|
2018-09-11 11:36:18 +00:00
|
|
|
include("layers/stateless.jl")
|
|
|
|
|
include("layers/conv.jl")
|
2018-09-05 14:39:00 +00:00
|
|
|
|
2019-11-04 14:37:25 +00:00
|
|
|
if Flux.use_cuda[]
|
2018-08-11 12:53:47 +00:00
|
|
|
include("cuda/cuda.jl")
|
2019-08-27 07:33:15 +00:00
|
|
|
else
|
|
|
|
|
@warn "CUDA unavailable, not testing GPU support"
|
2018-08-11 12:53:47 +00:00
|
|
|
end
|
2018-01-16 17:58:14 +00:00
|
|
|
|
2019-09-10 14:26:51 +00:00
|
|
|
if VERSION >= v"1.2"
|
|
|
|
|
doctest(Flux)
|
|
|
|
|
end
|
2019-09-10 14:02:43 +00:00
|
|
|
|
2017-07-04 17:32:21 +00:00
|
|
|
end
|
