diff options
author | Qi Liu <liuq901@163.com> | 2016-03-17 10:46:43 +0800 |
---|---|---|
committer | Qi Liu <liuq901@163.com> | 2016-03-17 10:46:43 +0800 |
commit | f1656ade78ac71d4cd84f1f7a539d1845b137589 (patch) | |
tree | b1535ca2f38edcc1f3af2559593732cd35d8db45 /nerv/layer/graph.lua | |
parent | e15307f071813e2eb56f7f83229b91141961325a (diff) |
RNN support multiple input & fix network flush bug
Diffstat (limited to 'nerv/layer/graph.lua')
-rw-r--r-- | nerv/layer/graph.lua | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/nerv/layer/graph.lua b/nerv/layer/graph.lua index 68d5f51..ddbc85e 100644 --- a/nerv/layer/graph.lua +++ b/nerv/layer/graph.lua @@ -65,7 +65,17 @@ function GraphLayer:discover(id, layer_repo) return ref end +local function reverse(connections) + for i = 1, #connections do + connections[i][3] = connections[i][3] * -1 + end +end + function GraphLayer:graph_init(layer_repo, connections) + if self.lconf.reversed then + reverse(connections) + end + local layers = {} layers['<input>'] = { inputs = {}, |