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

update for arrays

This commit is contained in:
Mike J Innes 2019-04-05 17:17:50 +01:00 committed by Elliot Saba
parent c70276ddfe
commit 92ddc618f8
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/optimise/train.jl
View File

@ -1,6 +1,11 @@
using Juno
import Zygote: Params, gradient
function update!(x::AbstractArray, x̄)
x .+= x̄
return x
end
function update!(opt, x, x̄)
update!(x, -apply!(opt, x, x̄))
end