CMakeLists.txt 652 字节
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 12 13
    optimizer.cc
    parameter_optimizer.cc
    sgd_optmizer.cc
    regularizer.cc
  )

set(OPITMIZER_Headers
14 15 16
    # adadelta_optimizer.h
    # adagrad_optimizer.h
    # adam_optimizer.h
17 18 19 20 21 22 23 24 25 26 27 28 29
    lr_policy.h
    optimizer.h
    parameter_optimizer.h
    regularizer.h
    sgd_optimizer.h
    Tensor.h
  )

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

add_simple_unittest(optimizer_test)
add_simple_unittest(optimizer_factory_test)
D
dzhwinter 已提交
30
add_simple_unittest(Tensor_test)