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

Updating pipeline tuner file

parent c7d76a62
No related branches found
No related tags found
No related merge requests found
......@@ -27,14 +27,19 @@ void testPipeline(){
// FIXIT: Extend this to batch of images - currently 5 images
//long int test_batch_size = 9145;
long int test_batch_size = 2000;
long int test_batch_size = 4572;
long int H = 240;
long int W = 300;
printf("Reading input\n");
void* input = readTrainedWeights("../model_params/pipeline/dataset/caltech101_255_float32.bin",
/*void* input = readTrainedWeights("../model_params/pipeline/dataset/caltech101_255_float32.bin",
float_type,
test_batch_size, 1, H, W);
test_batch_size, 1, H, W); */
void* input = readTrainedWeights("../model_params/pipeline/dataset/calibration_4572.bin",
float_type,
test_batch_size, 1, H, W);
printf("Reading golden output\n");
void* golden_output = readTrainedWeights("../model_params/pipeline/golden_output/GEOM-2000.bin",
//void* golden_output = readTrainedWeights("../model_params/pipeline/golden_output/caltech-gaussian.bin",
......@@ -114,12 +119,9 @@ void testPipeline(){
// NOTE-IMP: Important to include this call always before doing dumpOutput and computePSNViolation
hpvm_request_tensor(result, 0);
dumpOutput(result);
dumpOutput(result, "GEOM_calib.bin");
//void* psnr_output = readTrainedWeights("../model_params/pipeline/golden_output/caltech-gaussian.bin",
//float_type,
//test_batch_size, 1, H, W);
computePSNRViolation(result, golden_output, 30);
//computePSNRViolation(result, golden_output, 30);
dumpAccuracyNorms();
......
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