https://github.com/JuliaDiffEq/DiffEqFlux.jl
Tip revision: 4cc55daf3d83f13dd984d3f22d37f973d2ce669d authored by Christopher Rackauckas on 25 January 2019, 14:38:03 UTC
Merge pull request #13 from JuliaDiffEq/reverse_mode
Merge pull request #13 from JuliaDiffEq/reverse_mode
Tip revision: 4cc55da
utils.jl
using Flux, Test
using DiffEqFlux: destructure, restructure
model = Chain(Dense(10, 5, relu), Dense(5, 2))
p = destructure(model)
m2 = restructure(model, p)
@test !Tracker.isleaf(m2[1].W)
x = rand(10)
@test model(x) == m2(x)