summaryrefslogtreecommitdiff
path: root/embedding_example/setup_nerv.lua
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-08-06 14:08:26 +0800
committerDeterminant <[email protected]>2015-08-06 14:08:26 +0800
commitb4d9cfa8e3a4735687311577dded97d889340134 (patch)
tree49e0f000719705c563a357e0b89d62a66c84ce75 /embedding_example/setup_nerv.lua
parent2dc87bc02a1242dd5e029d0baaf4e0ae7173184f (diff)
make network configuration example file clearer
Diffstat (limited to 'embedding_example/setup_nerv.lua')
-rw-r--r--embedding_example/setup_nerv.lua5
1 files changed, 2 insertions, 3 deletions
diff --git a/embedding_example/setup_nerv.lua b/embedding_example/setup_nerv.lua
index 49a5dd6..5ade950 100644
--- a/embedding_example/setup_nerv.lua
+++ b/embedding_example/setup_nerv.lua
@@ -4,9 +4,8 @@ local arg = {...}
dofile(arg[1])
local param_repo = nerv.ParamRepo()
param_repo:import(gconf.initialized_param, nil, gconf)
-local sublayer_repo = make_sublayer_repo(param_repo)
-local layer_repo = make_layer_repo(sublayer_repo, param_repo)
-local network = get_network(layer_repo)
+local layer_repo = make_layer_repo(param_repo)
+local network = get_decode_network(layer_repo)
local global_transf = get_global_transf(layer_repo)
local batch_size = 1
network:init(batch_size)