diff options
| author | Ted Yin <ted.sybil@gmail.com> | 2015-10-12 09:25:32 +0800 |
|---|---|---|
| committer | Ted Yin <ted.sybil@gmail.com> | 2015-10-12 09:25:32 +0800 |
| commit | 64fce92b7845b716f3c168036691c37b2467d99b (patch) | |
| tree | 7c3da4cd2647d97cb9d1edb3386f59f64eb8030d /nerv/layer/affine.lua | |
| parent | bd6d0d8b72ec656dd8fa0c13aa602f4f6e022391 (diff) | |
| parent | 473eb9c082224be19f147697ba951ae5bac4b4b4 (diff) | |
Merge pull request #6 from yimmon/master
support batch_resize (sequence training related layers are in kaldi_seq)
Diffstat (limited to 'nerv/layer/affine.lua')
| -rw-r--r-- | nerv/layer/affine.lua | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua index b81b2a4..015ec3f 100644 --- a/nerv/layer/affine.lua +++ b/nerv/layer/affine.lua @@ -60,6 +60,10 @@ function AffineLayer:init(batch_size) self.bp:train_init() end +function AffineLayer:batch_resize(batch_size) + -- do nothing +end + function AffineLayer:update(bp_err, input, output) if self.direct_update then self.ltp.correction:mul(input[1], bp_err[1], 1.0, gconf.momentum, 'T', 'N') |