rmontanana/PyWrap
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add torch library

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-10-31 10:07:24 +01:00
parent 77c33942f6
commit cb3281ed91
16 changed files with 884 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/CMakeLists.txt
View File

@@ -1,6 +1,8 @@
include_directories(${PyWrap_SOURCE_DIR}/lib/Files)
include_directories(${Python3_INCLUDE_DIRS})
add_executable(main main.cc STree.cc SVC.cc PyClassifier.cc PyWrap.cc)
add_executable(example example.cpp)
target_link_libraries(main ${Python3_LIBRARIES} "${TORCH_LIBRARIES}")
target_link_libraries(main ${Python3_LIBRARIES} "${TORCH_LIBRARIES}" ArffFiles)
target_link_libraries(example "${TORCH_LIBRARIES}" ArffFiles)