From 462d1982f299c8be4ae9a746e6ae7d04c04faa33 Mon Sep 17 00:00:00 2001
From: Determinant <ted.sybil@gmail.com>
Date: Tue, 4 Aug 2015 14:41:22 +0800
Subject: add embedding_example

---
 embedding_example/setup_nerv.lua | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 embedding_example/setup_nerv.lua

(limited to 'embedding_example/setup_nerv.lua')

diff --git a/embedding_example/setup_nerv.lua b/embedding_example/setup_nerv.lua
new file mode 100644
index 0000000..e33a1e7
--- /dev/null
+++ b/embedding_example/setup_nerv.lua
@@ -0,0 +1,26 @@
+package.path="/home/slhome/mfy43/.luarocks/share/lua/5.1/?.lua;/home/slhome/mfy43/.luarocks/share/lua/5.1/?/init.lua;/home/slhome/mfy43/nerv/install/share/lua/5.1/?.lua;/home/slhome/mfy43/nerv/install/share/lua/5.1/?/init.lua;"..package.path
+package.cpath="/home/slhome/mfy43/.luarocks/lib/lua/5.1/?.so;/home/slhome/mfy43/nerv/install/lib/lua/5.1/?.so;"..package.cpath
+local k,l,_=pcall(require,"luarocks.loader") _=k and l.add_context("nerv","scm-1")
+
+local args = {...}
+require 'nerv'
+dofile(args[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 batch_size = 1
+network:init(batch_size)
+function propagator(input, output)
+    local gpu_input = nerv.CuMatrixFloat(input:nrow(), input:ncol())
+    local gpu_output = nerv.CuMatrixFloat(output:nrow(), output:ncol())
+    gpu_input:copy_fromh(input)
+    print(gpu_input)
+    network:propagate({gpu_input}, {gpu_output})
+    gpu_output:copy_toh(output)
+    print(output)
+    -- collect garbage in-time to save GPU memory
+    collectgarbage("collect")
+end
+return network.dim_in[1], network.dim_out[1], propagator
-- 
cgit v1.2.3-70-g09d2