diff options
Diffstat (limited to 'examples/test_dnn_layers.lua')
| -rw-r--r-- | examples/test_dnn_layers.lua | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/examples/test_dnn_layers.lua b/examples/test_dnn_layers.lua index c57de6d..866e685 100644 --- a/examples/test_dnn_layers.lua +++ b/examples/test_dnn_layers.lua @@ -2,11 +2,12 @@ require 'layer.affine' require 'layer.sigmoid' require 'layer.softmax_ce' -global_conf = {lrate = 0.8, wcost = 1e-6, momentum = 0.9} +global_conf = {lrate = 0.8, wcost = 1e-6, + momentum = 0.9, mat_type = nerv.CuMatrixFloat} -pf = nerv.ParamFile("affine.param", "r") -ltp = pf:read_param("a") -bp = pf:read_param("b") +pf = nerv.ChunkFile("affine.param", "r") +ltp = pf:read_chunk("a", global_conf) +bp = pf:read_chunk("b", global_conf) -- print(bp.trans) @@ -18,7 +19,7 @@ af:init() sg:init() sm:init() -df = nerv.ParamFile("input.param", "r") +df = nerv.ChunkFile("input.param", "r") label = nerv.CuMatrixFloat(10, 2048) label:fill(0) @@ -26,7 +27,7 @@ for i = 0, 9 do label[i][i] = 1.0 end -input1 = {[0] = df:read_param("input").trans} +input1 = {[0] = df:read_chunk("input", global_conf).trans} output1 = {[0] = nerv.CuMatrixFloat(10, 2048)} input2 = output1 output2 = {[0] = nerv.CuMatrixFloat(10, 2048)} |