From 121af0579ac662bb927d62bc40238655ae949b64 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Thu, 27 Apr 2017 17:27:08 +0100 Subject: [PATCH] update for rnn wrappers --- src/model.jl | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/model.jl b/src/model.jl index 371e7e2f..bfce352f 100644 --- a/src/model.jl +++ b/src/model.jl @@ -146,6 +146,8 @@ function back!(m::Stateful, Δ, x) back!(m.model, ((zeros.(m.ostate)...,), Δ), (m.istate...,), x)[2:end] end +update!(m::Stateful, η) = update!(m.model, η) + stateless(m) = m stateless(m::Stateful) = m.model @@ -166,3 +168,5 @@ function (m::SeqModel)(x) end back!(m::SeqModel, Δ, x) = (runseq((Δ, x) -> back!(m.model, Δ, x)[1], Δ, x),) + +update!(m::SeqModel, η) = update!(m.model, η)