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/nn/network.lua | |
parent | c682dfee8686c43aed8628633412c9b4d2bd708b (diff) |
update add_prefix for graph layer
Diffstat (limited to 'nerv/nn/network.lua')
-rw-r--r-- | nerv/nn/network.lua | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/nerv/nn/network.lua b/nerv/nn/network.lua index 0bbcc59..39df5f0 100644 --- a/nerv/nn/network.lua +++ b/nerv/nn/network.lua @@ -18,7 +18,8 @@ function network:__init(id, global_conf, network_conf) self.layers = {} self.input_conn = {} self.output_conn = {} - self.socket = self:compile(network_conf.network) + self.network = network_conf.network + self.socket = self:compile(self.network) for i = 1, #self.dim_in do local edge = self.socket.inputs[i] local id, port, time = edge[1], edge[2], edge[3] @@ -472,3 +473,15 @@ function network:update(bp_err, input, output) end end end + +function network:set_attr(name, value) + self.network:set_attr(name, value) +end + +function network:get_sublayer(id) + return self.network:get_sublayer(id) +end + +function network:get_params() + return self.network:get_params() +end |