From d2afc51a4e1a13fc3ad846e18c8ac098e3cd6bc7 Mon Sep 17 00:00:00 2001 From: Jonathan Strandberg Date: Fri, 19 Mar 2021 10:31:18 +0100 Subject: Add script to simplify testing networks on model The script will build the platform, optimize the network through vela, run it on the model and present relevant statistics from the run. Change-Id: I6bf7d77caa68e64c52f7b523e6b79af9271b6500 --- applications/baremetal/main.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'applications/baremetal/main.cpp') diff --git a/applications/baremetal/main.cpp b/applications/baremetal/main.cpp index e1e01de..4c0bf50 100644 --- a/applications/baremetal/main.cpp +++ b/applications/baremetal/main.cpp @@ -47,7 +47,7 @@ __attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t TFLuTensorAre InferenceProcess::InferenceProcess inferenceProcess(TFLuTensorArena, TENSOR_ARENA_SIZE); -uint8_t outputData[176] __attribute__((aligned(16), section("output_data_sec"))); +uint8_t outputData[sizeof(expectedOutputData)] __attribute__((aligned(16), section("output_data_sec"))); int runInference() { // Load inference data @@ -81,4 +81,4 @@ int runInference() { int main() { int ret = runInference(); return ret; -} \ No newline at end of file +} -- cgit v1.2.1