diff options
author | Determinant <ted.sybil@gmail.com> | 2016-03-16 17:53:39 +0800 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2016-03-16 17:53:39 +0800 |
commit | 289ac7f4b6e88b935da5c891e1efcf91fc047403 (patch) | |
tree | d4fc3a4fc20f2d5908624b3f6587ecd57966d719 /nerv/io/seq_buffer.lua | |
parent | 07fc1e2794027d44c255e1062c4491346b101a08 (diff) |
merge seq_buffer and change asr_trainer.lua accordingly
Diffstat (limited to 'nerv/io/seq_buffer.lua')
-rw-r--r-- | nerv/io/seq_buffer.lua | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/nerv/io/seq_buffer.lua b/nerv/io/seq_buffer.lua index ad1b3f7..029e7b8 100644 --- a/nerv/io/seq_buffer.lua +++ b/nerv/io/seq_buffer.lua @@ -5,7 +5,10 @@ function SeqBuffer:__init(global_conf, buffer_conf) self.batch_size = buffer_conf.batch_size self.chunk_size = buffer_conf.chunk_size - self.readers = buffer_conf.readers + self.readers = {} + for _, v in ipairs(buffer_conf.readers) do + table.insert(self.readers, v.reader) + end self.nn_act_default = buffer_conf.nn_act_default if self.nn_act_default == nil then self.nn_act_default = 0 @@ -29,7 +32,7 @@ function SeqBuffer:new_mini_batch() end function SeqBuffer:saturate(batch) - if self.queue[self.head] ~= nil and self.queue[self.head].seq_length[batch] ~= 0 then + if self.queue[self.head] ~= nil and self.queue[self.head].seq_length[batch] ~= 0 then return true end local data = {} |