diff options
Diffstat (limited to 'nerv/examples/network_debug/network.lua')
-rw-r--r-- | nerv/examples/network_debug/network.lua | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/nerv/examples/network_debug/network.lua b/nerv/examples/network_debug/network.lua index 1841d21..386c3b0 100644 --- a/nerv/examples/network_debug/network.lua +++ b/nerv/examples/network_debug/network.lua @@ -72,12 +72,12 @@ function nn:process(data, do_train, reader) for t = 1, self.gconf.chunk_size do local tmp = info.output[t][1]:new_to_host() for i = 1, self.gconf.batch_size do - if t <= info.seq_length[i] then - total_err = total_err + math.log10(math.exp(tmp[i - 1][0])) - total_frame = total_frame + 1 - end + total_err = total_err + math.log10(math.exp(tmp[i - 1][0])) end end + for i = 1, self.gconf.batch_size do + total_frame = total_frame + info.seq_length[i] + end timer:toc('IO') timer:tic('network') |