github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-08 05:24:00 -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

14
cpp/vulkan_gui/vulkan_inference_gui.cc
View File

@@ -723,7 +723,7 @@ namespace iree {
extern "C" int iree_main(int argc, char** argv) {
fprintf(stdout, "starting yo\n");
// --------------------------------------------------------------------------
// Create a window.
if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_TIMER) != 0) {
@@ -747,7 +747,7 @@ extern "C" int iree_main(int argc, char** argv) {
abort();
return 1;
}
// Setup Vulkan
iree_hal_vulkan_features_t iree_vulkan_features =
static_cast<iree_hal_vulkan_features_t>(
@@ -898,10 +898,10 @@ extern "C" int iree_main(int argc, char** argv) {
iree_allocator_system(), &hal_module));
// Load bytecode module
// Load bytecode module
iree_file_toc_t module_file_toc;
const char network_model[] = "amd-resnet50.vmfb";
fprintf(stdout, "Loading: %s\n", network_model);
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)
{
abort();
@@ -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},
@@ -975,7 +975,7 @@ extern "C" int iree_main(int argc, char** argv) {
iree_string_view_t main_function_name = iree_vm_function_name(&main_function);
fprintf(stdout, "Resolved main function named '%.*s'\n",
(int)main_function_name.size, main_function_name.data);
// --------------------------------------------------------------------------
// Write inputs into mappable buffers.