diff options
| author | Determinant <[email protected]> | 2015-06-05 10:58:57 +0800 |
|---|---|---|
| committer | Determinant <[email protected]> | 2015-06-05 10:58:57 +0800 |
| commit | df737041e4a9f3f55978cc74db9a9cea27fa9fa0 (patch) | |
| tree | d656820be286550bc548f7c5ed4b1dcfecf3691c /examples/test_nn_lib.lua | |
| parent | ea6f2990f99dd9ded6a0e74d75a3ec84900a2518 (diff) | |
add profiling; add ce accurarcy; several other changes
Diffstat (limited to 'examples/test_nn_lib.lua')
| -rw-r--r-- | examples/test_nn_lib.lua | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/examples/test_nn_lib.lua b/examples/test_nn_lib.lua index 9600917..04fd7d6 100644 --- a/examples/test_nn_lib.lua +++ b/examples/test_nn_lib.lua @@ -116,7 +116,8 @@ tnet_reader = nerv.TNetReader(gconf, buffer = nerv.SGDBuffer(gconf, { - buffer_size = 8192, + buffer_size = 81920, + -- randomize = true, readers = { { reader = tnet_reader, data = {main_scp = 429, ref = 1}} @@ -126,10 +127,11 @@ buffer = nerv.SGDBuffer(gconf, sm = sublayer_repo:get_layer("softmax_ce0") main = layer_repo:get_layer("main") main:init(gconf.batch_size) -cnt = 0 +gconf.cnt = 0 for data in buffer.get_data, buffer do - if cnt == 1000 then break end - cnt = cnt + 1 + if gconf.cnt == 1000 then break end + gconf.cnt = gconf.cnt + 1 + input = {data.main_scp, data.ref} output = {} err_input = {} @@ -140,7 +142,10 @@ for data in buffer.get_data, buffer do main:update(err_input, input, output) nerv.utils.printf("cross entropy: %.8f\n", sm.total_ce) - nerv.utils.printf("frames: %.8f\n", sm.total_frames) + nerv.utils.printf("correct: %d\n", sm.total_correct) + nerv.utils.printf("frames: %d\n", sm.total_frames) nerv.utils.printf("err/frm: %.8f\n", sm.total_ce / sm.total_frames) + nerv.utils.printf("accuracy: %.8f\n", sm.total_correct / sm.total_frames) collectgarbage("collect") end +nerv.Matrix.print_profile() |