diff options
author | Determinant <[email protected]> | 2016-02-16 17:04:44 +0800 |
---|---|---|
committer | Determinant <[email protected]> | 2016-02-16 17:04:44 +0800 |
commit | 7dee5871f8f67a78ee704c9efd5d4708e8a27740 (patch) | |
tree | 19d672ba48d4805f3fe7e5c8ba47936dd8a2225c /nerv/nn | |
parent | 9e7171e2da3e4edba303f5c2bdaef416fb62e81a (diff) |
improve parameter auto-detection
Diffstat (limited to 'nerv/nn')
-rw-r--r-- | nerv/nn/layer_repo.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nerv/nn/layer_repo.lua b/nerv/nn/layer_repo.lua index ef333a7..2f8de08 100644 --- a/nerv/nn/layer_repo.lua +++ b/nerv/nn/layer_repo.lua @@ -24,6 +24,9 @@ function LayerRepo:add_layers(layer_spec, param_repo, global_conf) for pname, pid in pairs(spec[1]) do layer_config[pname] = param_repo:get_param(pid) end + if layer_config.pr == nil then + layer_config.pr = param_repo + end layers[id] = layer_type(id, global_conf, layer_config) end end |