diff options
| author | Determinant <ted.sybil@gmail.com> | 2016-02-15 16:18:13 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2016-02-15 16:18:13 +0800 |
| commit | a7608a67f16f745309409f6a004354121e0b3ba6 (patch) | |
| tree | b36ec7c4e8b252f540f049d374a13915ca5ba9a3 /nerv/layer | |
| parent | 3362020a6bc43766a92882abe6d127c8bb98a628 (diff) | |
move curand_gen out of the matrix layout
Diffstat (limited to 'nerv/layer')
| -rw-r--r-- | nerv/layer/affine.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua index 0d4f7dd..ec13519 100644 --- a/nerv/layer/affine.lua +++ b/nerv/layer/affine.lua @@ -80,6 +80,9 @@ function AffineLayer:__init(id, global_conf, layer_conf) self.id = id self.dim_in = layer_conf.dim_in self.dim_out = layer_conf.dim_out + if layer_conf.ltp ~= nil and layer_conf.ltp1 == nil then + layer_conf.ltp1 = layer_conf.ltp + end for i = 1, #self.dim_in do self["ltp" .. i] = self:find_param("ltp" .. i, layer_conf, global_conf, nerv.LinearTransParam, |