aboutsummaryrefslogtreecommitdiff
path: root/Makefile
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-05-28 17:01:10 +0800
committerDeterminant <[email protected]>2015-05-28 17:01:10 +0800
commite934b616496940bfe0924ca1992035d2346baa62 (patch)
tree6ed5398d9123cc2cbfd2b09ac1aed74db42299c4 /Makefile
parente4dedc2992149d245ea65132131253072d3276b8 (diff)
add softmax + ce layer; test_dnn_layers produces the same result as TNet
Diffstat (limited to 'Makefile')
-rw-r--r--Makefile2
1 files changed, 1 insertions, 1 deletions
diff --git a/Makefile b/Makefile
index 727765d..d403e8f 100644
--- a/Makefile
+++ b/Makefile
@@ -6,7 +6,7 @@ OBJS := nerv.o luaT.o common.o \
LIBS := libnerv.so
LUA_LIBS := matrix/init.lua io/init.lua nerv.lua \
pl/utils.lua pl/compat.lua \
- layer/init.lua layer/affine.lua layer/sigmoid.lua
+ layer/init.lua layer/affine.lua layer/sigmoid.lua layer/softmax_ce.lua
INCLUDE := -I build/luajit-2.0/include/luajit-2.0/ -DLUA_USE_APICHECK
CUDA_BASE := /usr/local/cuda-6.5
CUDA_INCLUDE := -I $(CUDA_BASE)/include/