Skip to content
Snippets Groups Projects
Commit fce8e32e authored by Hashim Sharif's avatar Hashim Sharif
Browse files

Fixing ResNet18_cifar10 paths

parent 7c5414d9
No related branches found
No related tags found
No related merge requests found
Showing
with 6 additions and 6 deletions
......@@ -13,7 +13,7 @@ int main(){
llvm_hpvm_initTensorRt(0);
std::string dir_prefix = std::string("../model_params/resnet18_cifar10_promise/");
std::string dir_prefix = std::string("../model_params/resnet18_cifar10/");
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");
......
......@@ -17,7 +17,7 @@ int to_skip = 5;
int main(int argc, char* argv[]){
int test_input_size = 3000;
int test_input_size = 5000;
int batch_size = 1000;
int offset = 5000;
......@@ -54,11 +54,11 @@ int main(int argc, char* argv[]){
classConfs = (float*) malloc(sizeof(float) * test_input_size);
predictedLabels = (int*) malloc(sizeof(int) * test_input_size);
}
llvm_hpvm_initTensorRt(1);
llvm_hpvm_initTensorRt(0);
int missed = 0;
for (int i = 0 ; i < total_runs; i++){
......@@ -72,7 +72,7 @@ int main(int argc, char* argv[]){
float final_accuracy = 0.0;
std::string dir_prefix = std::string("../model_params/resnet18_cifar10_promise/");
std::string dir_prefix = std::string("../model_params/resnet18_cifar10/");
std::string input_path = dir_prefix + std::string("input.bin");
std::string labels_path = dir_prefix + std::string("labels.bin");
std::string labels32_path = dir_prefix + std::string("labels32.bin");
......
......@@ -13,7 +13,7 @@ int main(){
llvm_hpvm_initTensorRt(1);
std::string dir_prefix = std::string("../model_params/resnet18_cifar10_promise/");
std::string dir_prefix = std::string("../model_params/resnet18_cifar10/");
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");
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment