diff options
| author | Determinant <ted.sybil@gmail.com> | 2016-02-16 17:04:44 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2016-02-16 17:04:44 +0800 |
| commit | 7dee5871f8f67a78ee704c9efd5d4708e8a27740 (patch) | |
| tree | 19d672ba48d4805f3fe7e5c8ba47936dd8a2225c /nerv/nn/layer_repo.lua | |
| parent | 9e7171e2da3e4edba303f5c2bdaef416fb62e81a (diff) | |
improve parameter auto-detection
Diffstat (limited to 'nerv/nn/layer_repo.lua')
| -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 |