From d7d399a5022d850998d833c6859cd3be24a6b958 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu>
Date: Sat, 13 Jun 2020 17:01:01 -0500
Subject: [PATCH] Fixing input data path in ResNet18 script
---
.../projects/hpvm-tensor-rt/dnn_sources/src/resnet18_cifar10.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/resnet18_cifar10.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/resnet18_cifar10.cc
index dcc095c254..bc3022234a 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/resnet18_cifar10.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/resnet18_cifar10.cc
@@ -14,7 +14,7 @@ int main(){
llvm_hpvm_initTensorRt(1);
std::string dir_prefix = std::string("../model_params/resnet18_cifar10_promise/");
- std::string input_path = dir_prefix + std::string("extended_input.bin");
+ std::string input_path = dir_prefix + std::string("input.bin");
//void* input = readTrainedWeights(input_path.c_str(), 0, batch_size,3,32,32);
std::string labels_path = dir_prefix + std::string("labels.bin");
//uint8_t* labels = readLabels(labels_path.c_str(), batch_size);
--
GitLab