update add_prefix for graph layer

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