aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-10-09 18:42:09 +0800
committerDeterminant <[email protected]>2015-10-09 18:42:09 +0800
commitbd6d0d8b72ec656dd8fa0c13aa602f4f6e022391 (patch)
tree2ff5f55c1b813c7fc1a043252070ee6c5f08adba
parent3721c74d56ffdea43851489617f33cd13b87ab76 (diff)
enable user to set direct_update in gconf; use direct_update in the example
-rw-r--r--nerv/examples/swb_baseline.lua1
-rw-r--r--nerv/layer/affine.lua2
2 files changed, 2 insertions, 1 deletions
diff --git a/nerv/examples/swb_baseline.lua b/nerv/examples/swb_baseline.lua
index 8015884..8f72200 100644
--- a/nerv/examples/swb_baseline.lua
+++ b/nerv/examples/swb_baseline.lua
@@ -2,6 +2,7 @@ require 'htk_io'
gconf = {lrate = 0.8, wcost = 1e-6, momentum = 0.9,
cumat_type = nerv.CuMatrixFloat,
mmat_type = nerv.MMatrixFloat,
+ direct_update = true,
frm_ext = 5,
frm_trim = 5,
tr_scp = "/slfs1/users/mfy43/swb_ivec/train_bp.scp",
diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua
index 00cbcfb..b81b2a4 100644
--- a/nerv/layer/affine.lua
+++ b/nerv/layer/affine.lua
@@ -42,7 +42,7 @@ function AffineLayer:__init(id, global_conf, layer_conf)
self.dim_out = layer_conf.dim_out
self.gconf = global_conf
self:check_dim_len(1, 1) -- exactly one input and one output
- self.direct_update = layer_conf.direct_update
+ self.direct_update = layer_conf.direct_update or global_conf.direct_update
end
function AffineLayer:init(batch_size)