added has_param api for param_repo

1 files changed, 13 insertions, 2 deletions

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