Flux.jl/test/runtests.jl

34 lines
546 B
Julia
Raw Normal View History

2018-09-11 11:28:05 +00:00
using Flux, Test, Random, Statistics
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
2018-08-20 12:08:04 +00:00
# So we can use the system CuArrays
insert!(LOAD_PATH, 2, "@v#.#")
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
@info "Running Gradient Checks"
2018-09-11 11:36:18 +00:00
include("tracker.jl")
2017-06-26 09:21:17 +00:00
2018-08-20 12:08:04 +00:00
if Base.find_package("CuArrays") != nothing
2018-08-11 12:53:47 +00:00
include("cuda/cuda.jl")
end
2018-01-16 17:58:14 +00:00
2017-07-04 17:32:21 +00:00
end