From e934b616496940bfe0924ca1992035d2346baa62 Mon Sep 17 00:00:00 2001 From: Determinant Date: Thu, 28 May 2015 17:01:10 +0800 Subject: add softmax + ce layer; test_dnn_layers produces the same result as TNet --- Makefile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'Makefile') 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/ -- cgit v1.2.3