CMakeLists.txt 622 字节
Newer Older
1 2 3
include_directories(${CMAKE_CURRENT_BINARY_DIR})

set(OPITMIZER_SRCS
4 5 6
    adadelta_optimizer.cc
    adagrad_optimizer.cc
    adam_optimizer.cc
7 8 9 10 11
    optimizer.cc
    parameter_optimizer.cc
    sgd_optmizer.cc
  )

D
dzhwinter 已提交
12
set(OPITMIZER_HEADERS
13 14 15
    adadelta_optimizer.h
    adagrad_optimizer.h
    adam_optimizer.h
16 17 18 19
    lr_policy.h
    optimizer.h
    parameter_optimizer.h
    sgd_optimizer.h
D
dzhwinter 已提交
20
    tensor.h
21 22 23 24 25
  )

add_library(optimizer STATIC ${OPITMIZER_SRCS})
add_dependencies(optimizer gen_proto_cpp)

D
dzhwinter 已提交
26
add_simple_unittest(tensor_test)
D
dzhwinter 已提交
27 28
add_simple_unittest(parameter_optimizer_test)
add_dependencies(parameter_optimizer_test optimizer)