use explicit update! rule

This commit is contained in:
Dhairya Gandhi 2018-10-27 19:39:56 +05:30
parent 815e8c206d
commit ea508a79b0
1 changed files with 1 additions and 1 deletions

View File

@ -16,7 +16,7 @@ using Test
for t = 1: 10^5
l = loss(rand(10))
back!(l)
delta = Optimise.update!(opt, w)
delta = Optimise.update!(opt, w.data, w.grad)
w.data .-= delta
end
@test Flux.mse(w, w) < 0.01