mirror of
https://github.com/marian-nmt/marian.git
synced 2024-09-17 09:47:34 +03:00
Added encoder-decoder validator.
This commit is contained in:
parent
8dc655dad4
commit
98d66f738a
@ -34,11 +34,16 @@ cuda_add_executable(
|
|||||||
validate_mnist_batch
|
validate_mnist_batch
|
||||||
validate_mnist_batch.cu
|
validate_mnist_batch.cu
|
||||||
)
|
)
|
||||||
|
cuda_add_executable(
|
||||||
|
validate_encoder_decoder
|
||||||
|
validate_encoder_decoder.cu
|
||||||
|
)
|
||||||
|
|
||||||
target_link_libraries(validate_mnist marian_lib)
|
target_link_libraries(validate_mnist marian_lib)
|
||||||
target_link_libraries(validate_mnist_batch marian_lib)
|
target_link_libraries(validate_mnist_batch marian_lib)
|
||||||
|
target_link_libraries(validate_encoder_decoder marian_lib)
|
||||||
|
|
||||||
foreach(exec marian train_mnist validate_mnist validate_mnist_batch )
|
foreach(exec marian train_mnist validate_mnist validate_mnist_batch validate_encoder_decoder)
|
||||||
target_link_libraries(${exec} ${EXT_LIBS} cuda cudnn)
|
target_link_libraries(${exec} ${EXT_LIBS} cuda cudnn)
|
||||||
cuda_add_cublas_to_target(${exec})
|
cuda_add_cublas_to_target(${exec})
|
||||||
set_target_properties(${exec} PROPERTIES RUNTIME_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}")
|
set_target_properties(${exec} PROPERTIES RUNTIME_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}")
|
||||||
|
Loading…
Reference in New Issue
Block a user