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

Project.toml and Manifest updated

This commit is contained in:
thebhatman 2019-07-02 12:16:24 +05:30
parent 618f8a03c8
commit 9f6793d63a
4 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Manifest.toml
View File

@ -276,12 +276,6 @@ git-tree-sha1 = "3e83f60b74911d3042d3550884ca2776386a02b8"
uuid = "0796e94c-ce3b-5d07-9a54-7f471281c624" uuid = "0796e94c-ce3b-5d07-9a54-7f471281c624"
version = "0.5.3" version = "0.5.3"
[[Tracker]]
deps = ["Adapt", "DiffRules", "ForwardDiff", "LinearAlgebra", "MacroTools", "NNlib", "NaNMath", "Printf", "Random", "Requires", "SpecialFunctions", "Statistics", "Test"]
git-tree-sha1 = "0bec1b68c63a0e8a58d3944261cbf4cc9577c8a1"
uuid = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
version = "0.2.0"
[[TranscodingStreams]] [[TranscodingStreams]]
deps = ["Random", "Test"] deps = ["Random", "Test"]
git-tree-sha1 = "a25d8e5a28c3b1b06d3859f30757d43106791919" git-tree-sha1 = "a25d8e5a28c3b1b06d3859f30757d43106791919"

3
Project.toml
View File

@ -21,13 +21,12 @@ SHA = "ea8e919c-243c-51af-8825-aaa63cd721ce"
Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2" Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91" StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40" Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
ZipFile = "a5390f91-8eb1-5f08-bee0-b1d1ffed6cea" ZipFile = "a5390f91-8eb1-5f08-bee0-b1d1ffed6cea"
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
[compat] [compat]
NNlib = "0.6" NNlib = "0.6"
Tracker = "0.2" Zygote = "0.3"
julia = "0.7, 1" julia = "0.7, 1"
[extras] [extras]

2
test/runtests.jl
View File

@ -24,7 +24,7 @@ include("layers/conv.jl")
@info "Running Gradient Checks" @info "Running Gradient Checks"
include("tracker.jl") include("zygote.jl")
if Base.find_package("CuArrays") != nothing if Base.find_package("CuArrays") != nothing
include("cuda/cuda.jl") include("cuda/cuda.jl")

2
test/tracker.jl → test/zygote.jl
View File

@ -22,7 +22,7 @@ gradcheck(f, xs...) =
gradtest(f, xs::AbstractArray...) = gradcheck((xs...) -> sum(sin.(f(xs...))), xs...) gradtest(f, xs::AbstractArray...) = gradcheck((xs...) -> sum(sin.(f(xs...))), xs...)
gradtest(f, dims...) = gradtest(f, rand.(Float64, dims)...) gradtest(f, dims...) = gradtest(f, rand.(Float64, dims)...)
@testset "Tracker" begin @testset "Zygote" begin
@test gradtest(Flux.mse, rand(5,5), rand(5, 5)) @test gradtest(Flux.mse, rand(5,5), rand(5, 5))
@test gradtest(Flux.crossentropy, rand(5,5), rand(5, 5)) @test gradtest(Flux.crossentropy, rand(5,5), rand(5, 5))