Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[DRAFT][onert-micro] Onert-micro training PoCv3 #13107

Closed
wants to merge 8 commits into from
Closed

[DRAFT][onert-micro] Onert-micro training PoCv3 #13107

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
68d8c74
[DRAFT] Onert-micro training PoCv2
May 29, 2024
090def3
add usage without wof file and add save method to Driver.
Jun 4, 2024
8bb6635
remove circle divider
Jun 4, 2024
8e1774a
change Adam optimizer, and fix some small details
Jun 5, 2024
99ae4b4
move saving model into runtime api
Jun 5, 2024
39c479e
add checkpoints and small fix for adam optimizer
Jun 6, 2024
01d7bae
add tests
Jun 11, 2024
130afb4
rebase fixes
Jun 17, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 2 additions & 6 deletions onert-micro/eval-driver/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,19 +14,15 @@ install(TARGETS onert_micro_eval_driver DESTINATION bin)

message(STATUS "DONE eval driver")

if(NOT ENABLE_ONERT_MICRO_TRAINING)
return()
endif(NOT ENABLE_ONERT_MICRO_TRAINING)

set(SRCS_EVAL_TRAINING_TESTER TrainingDriver.cpp)

add_executable(onert_micro_training_eval_driver ${SRCS_EVAL_TRAINING_TESTER})

# This variable is needed to separate standalone interpreter libraries from the libraries used in driver
set(READER_SUFFIX "_driver")

target_include_directories(onert_micro_training_eval_driver PUBLIC "${CMAKE_CURRENT_SOURCE_DIR}/luci-interpreter/include")
target_link_libraries(onert_micro_training_eval_driver PUBLIC luci_interpreter_micro)
target_include_directories(onert_micro_training_eval_driver PUBLIC "${CMAKE_CURRENT_SOURCE_DIR}/onert_micro/include")
target_link_libraries(onert_micro_training_eval_driver PUBLIC onert_micro_training_interpreter)

install(TARGETS onert_micro_training_eval_driver DESTINATION bin)

Expand Down
Loading
Loading