From 74d9e9e7371c80394698fb9805cbf0cbde67a8f3 Mon Sep 17 00:00:00 2001
From: Determinant <ted.sybil@gmail.com>
Date: Tue, 2 Jun 2015 20:28:16 +0800
Subject: add ParamRepo, LayerRepo, DAGLayer

---
 nn/layer_repo.lua | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 nn/layer_repo.lua

(limited to 'nn/layer_repo.lua')

diff --git a/nn/layer_repo.lua b/nn/layer_repo.lua
new file mode 100644
index 0000000..b1d2248
--- /dev/null
+++ b/nn/layer_repo.lua
@@ -0,0 +1,34 @@
+local LayerRepo = nerv.class("nerv.LayerRepo")
+
+function LayerRepo:__init(layer_spec, param_repo, global_conf)
+    local layers = {}
+    for ltype, llist in pairs(layer_spec) do
+        local layer_type = nerv.get_type(ltype)
+        for id, spec in pairs(llist) do
+            if layers[id] ~= nil then
+                nerv.error("a layer with id %s already exists", id)
+            end
+            nerv.utils.printf("id: %s\n", id)
+            if type(spec[2]) ~= "table" then
+                nerv.error("layer config table is need")
+            end
+            layer_config = spec[2]
+            if type(spec[1]) ~= "table" then
+                nerv.error("parameter description table is needed")
+            end
+            for pname, pid in pairs(spec[1]) do
+                layer_config[pname] = param_repo:get_param(pid, global_conf)
+            end
+            layers[id] = layer_type(id, global_conf, layer_config)
+        end
+    end
+    self.layers = layers
+end
+
+function LayerRepo:get_layer(lid)
+    local layer = self.layers[lid]
+    if layer == nil then
+        nerv.error("layer with id %s not found", lid)
+    end
+    return layer
+end
-- 
cgit v1.2.3