diff options
| author | Determinant <ted.sybil@gmail.com> | 2015-05-28 17:01:10 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2015-05-28 17:01:10 +0800 |
| commit | e934b616496940bfe0924ca1992035d2346baa62 (patch) | |
| tree | 6ed5398d9123cc2cbfd2b09ac1aed74db42299c4 /Makefile | |
| parent | e4dedc2992149d245ea65132131253072d3276b8 (diff) | |
add softmax + ce layer; test_dnn_layers produces the same result as TNet
Diffstat (limited to 'Makefile')
| -rw-r--r-- | Makefile | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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/ |