added has_param api for param_repo

2 files changed, 13 insertions, 5 deletions

diff --git a/nerv/nn/layer_repo.lua b/nerv/nn/layer_repo.lua
index ec0f80a..ef333a7 100644
--- a/nerv/nn/layer_repo.lua
+++ b/nerv/nn/layer_repo.lua
@@ -23,9 +23,6 @@ function LayerRepo:add_layers(layer_spec, param_repo, global_conf)
             end
             for pname, pid in pairs(spec[1]) do
                 layer_config[pname] = param_repo:get_param(pid)
-                if layer_config[pname] == nil then
-                    nerv.error("did not find parameter in paramRepo")
-                end
             end
             layers[id] = layer_type(id, global_conf, layer_config)
         end
diff --git a/nerv/nn/param_repo.lua b/nerv/nn/param_repo.lua
index 7fc0498..6d52691 100644
--- a/nerv/nn/param_repo.lua
+++ b/nerv/nn/param_repo.lua
@@ -67,7 +67,18 @@ function ParamRepo:export(param_file, pids)
     cf:close()
 end
 
+function ParamRepo:has_param(pid)
+    if self.params[pid] ~= nil then
+        return true
+    else
+        return false
+    end
+end
+
 function ParamRepo:get_param(pid)
-    --if pid does not exist, return nil
-    return self.params[pid]
+    local p = self.params[pid]
+    if p == nil then
+        nerv.error("param with id %s not found", pid)
+    end
+    return p
 end