github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-08 21:38:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update resnet paths

Browse Source
This commit is contained in:
Anush Elangovan
2022-09-14 16:56:20 -07:00
parent 174b171913
commit 3b0ec8ce4e
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
cpp/vulkan_gui/dog_imagenet.jpg → cpp/dog_imagenet.jpg
View File

Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 26 KiB

After

Width:  |  Height:  |  Size: 26 KiB

4
cpp/vulkan_gui/vulkan_inference_gui.cc
View File

@@ -900,7 +900,7 @@ extern "C" int iree_main(int argc, char** argv) {
// Load bytecode module
iree_file_toc_t module_file_toc;
const char network_model[] = "amd-resnet50.vmfb";
const char network_model[] = "resnet50_tf.vmfb";
fprintf(stdout, "Loading: %s\n", network_model);
if (load_file(network_model, &module_file_toc.data, &module_file_toc.size) == false)
{
@@ -967,7 +967,7 @@ extern "C" int iree_main(int argc, char** argv) {
// Lookup the entry point function.
iree_vm_function_t main_function;
const char kMainFunctionName[] = "module.predict";
const char kMainFunctionName[] = "module.forward";
IREE_CHECK_OK(iree_vm_context_resolve_function(
iree_context,
iree_string_view_t{kMainFunctionName, sizeof(kMainFunctionName) - 1},