sdobber / FluxArchitectures.jl

Complex neural network examples for Flux.jl
MIT License
122 stars 15 forks source link

Fix Flux deprecations and example files #47

Closed sdobber closed 2 years ago

sdobber commented 2 years ago

Change ADAM to Adam, replace Flux.Zygote.@nograd with Flux.ChainRulesCore.@non_differentiable, and update the examples with

Flux.ChainRulesCore.ignore_derivatives() do
        Flux.reset!(model)
end

to avoid errors coming from differentiating through the reset!-call.