diff options
author | Qi Liu <liuq901@163.com> | 2016-03-03 19:42:15 +0800 |
---|---|---|
committer | Qi Liu <liuq901@163.com> | 2016-03-03 19:42:15 +0800 |
commit | 8374e8fbc545633b6adf5c4090af8997a65778d2 (patch) | |
tree | 9d959337628686b2b9ece9016a92ea55d40c0d31 /nerv/layer/rnn.lua | |
parent | c682dfee8686c43aed8628633412c9b4d2bd708b (diff) |
update add_prefix for graph layer
Diffstat (limited to 'nerv/layer/rnn.lua')
-rw-r--r-- | nerv/layer/rnn.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/nerv/layer/rnn.lua b/nerv/layer/rnn.lua index 806ac58..38f2326 100644 --- a/nerv/layer/rnn.lua +++ b/nerv/layer/rnn.lua @@ -27,8 +27,6 @@ function RNNLayer:__init(id, global_conf, layer_conf) } } - local layer_repo = nerv.LayerRepo(layers, pr, global_conf) - local connections = { {'<input>[1]', 'main[1]', 0}, {'main[1]', 'sigmoid[1]', 0}, @@ -37,5 +35,7 @@ function RNNLayer:__init(id, global_conf, layer_conf) {'dup[2]', '<output>[1]', 0}, } + self:add_prefix(layers, connections) + local layer_repo = nerv.LayerRepo(layers, pr, global_conf) self:graph_init(layer_repo, connections) end |