-
Notifications
You must be signed in to change notification settings - Fork 6
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.
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
Sciwork demo #6
base: master
Are you sure you want to change the base?
Sciwork demo #6
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the work!
One minor change requested in the main()
for readability.
Though, this is an example that's meant to be run on all devices. The inclusion of additional image samples might not fit on the tiniest devices. My recommendation is holding back this PR until tests have been conduced for cortex M0 devices.
// create the input/output tensor | ||
Tensor input_image = new RomTensor({1, 28, 28, 1}, flt, arr_input_image); | ||
Tensor logits = new RamTensor({1, 10}, flt); | ||
size_t num_samples = *(&ref_labels + 1) - ref_labels; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
A macro define or a const variable is a better idea here.
demo multiple inferences