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

use explicit update! rule

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/optimise.jl
View File

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