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

rename seqmodel

This commit is contained in:
Mike J Innes 2016-10-28 19:11:38 +01:00
parent 102e09d14b
commit 1c6eaece5d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/backend/tensorflow/recurrent.jl
View File

@ -1,4 +1,4 @@
immutable RModel
immutable SeqModel
m::Model
end
@ -11,7 +11,7 @@ function tf(model::Flux.Unrolled)
run(sess, initialize_all_variables())
Model(model, sess, params,
[input], [output],
[gradients(output, input)]) |> RModel
[gradients(output, input)]) |> SeqModel
end
function batchseq(xs)
@ -22,6 +22,6 @@ function batchseq(xs)
Batch{Seq{T,S},B}(xs)
end
(m::RModel)(x::BatchSeq) = batchseq(rawbatch(m.m(x)))
(m::SeqModel)(x::BatchSeq) = batchseq(rawbatch(m.m(x)))
(m::RModel)(x::Seq) = first(m(batchone(x)))
(m::SeqModel)(x::Seq) = first(m(batchone(x)))