regression test

2017-12-08 18:24:07 +00:00 · 2017-12-08 18:24:07 +00:00 · 55bbe50f32
parent 24a6569589
commit 55bbe50f32
1 changed files with 1 additions and 1 deletions
--- a/test/optimise.jl
+++ b/test/optimise.jl
@ -3,7 +3,7 @@ using Flux.Tracker

@testset "Optimise" begin
  w = randn(10, 10)
-  for Opt in [SGD, Nesterov, Momentum, ADAM, RMSProp, ps -> ADAGrad(ps, 0.1), ADADelta]
+  for Opt in [SGD, Nesterov, Momentum, ADAM, RMSProp, ps -> ADAGrad(ps, 0.1), ADADelta, AMSGrad]
    w′ = param(randn(10, 10))
    loss(x) = Flux.mse(w*x, w′*x)
    opt = Opt([w′])