diff options
author | Determinant <[email protected]> | 2016-02-15 16:18:13 +0800 |
---|---|---|
committer | Determinant <[email protected]> | 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, |