diff options
author | txh18 <[email protected]> | 2015-11-20 19:58:14 +0800 |
---|---|---|
committer | txh18 <[email protected]> | 2015-11-20 19:58:14 +0800 |
commit | 4f5b45b79b8d5f6a9094888cf6b929fe86ac24a3 (patch) | |
tree | 60d5b6232b8d33da9178717c27ecb6dc3591d0b6 /nerv/nn | |
parent | 6456f5d0b97c5ad7e35c58751f74b8c9fefb635e (diff) |
working on automatic parameter for layers
Diffstat (limited to 'nerv/nn')
-rw-r--r-- | nerv/nn/layer_repo.lua | 3 | ||||
-rw-r--r-- | nerv/nn/param_repo.lua | 7 |
2 files changed, 5 insertions, 5 deletions
diff --git a/nerv/nn/layer_repo.lua b/nerv/nn/layer_repo.lua index ef333a7..ec0f80a 100644 --- a/nerv/nn/layer_repo.lua +++ b/nerv/nn/layer_repo.lua @@ -23,6 +23,9 @@ 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 ab971ba..7fc0498 100644 --- a/nerv/nn/param_repo.lua +++ b/nerv/nn/param_repo.lua @@ -68,9 +68,6 @@ function ParamRepo:export(param_file, pids) end function ParamRepo:get_param(pid) - local p = self.params[pid] - if p == nil then - nerv.error("param with id %s not found", pid) - end - return p + --if pid does not exist, return nil + return self.params[pid] end |