From 0eb8a8a98da2941126221296d29949b6c10b23f9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Manuel=20Ramos=20Calder=C3=B3n?= Date: Tue, 21 Feb 2023 11:23:35 +0100 Subject: [PATCH 1/2] Fixed missing headers in test.sh PSNR and MSE are computed in test.sh and added in the .md but the headers are missing, giving the wrong information when visualizing the outputted .md --- test.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.sh b/test.sh index 3bf4a4d8..979b376c 100755 --- a/test.sh +++ b/test.sh @@ -4,7 +4,7 @@ OUTPUT_FILE=$1 touch $OUTPUT_FILE -echo "| Name | Data Type | Input Shapes | torch2trt kwargs | Max Error | Throughput (PyTorch) | Throughput (TensorRT) | Latency (PyTorch) | Latency (TensorRT) |" >> $OUTPUT_FILE +echo "| Name | Data Type | Input Shapes | torch2trt kwargs | Max Error | PSNR | MSE | Throughput (PyTorch) | Throughput (TensorRT) | Latency (PyTorch) | Latency (TensorRT) |" >> $OUTPUT_FILE echo "|------|-----------|--------------|------------------|-----------|----------------------|-----------------------|-------------------|--------------------|" >> $OUTPUT_FILE python3 -m torch2trt.test -o $OUTPUT_FILE --name alexnet --include=torch2trt.tests.torchvision.classification From 75365088dfe1d0aa3805a328038b9543cd7895cf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Manuel=20Ramos=20Calder=C3=B3n?= Date: Tue, 21 Feb 2023 11:39:46 +0100 Subject: [PATCH 2/2] minor table trailing --- test.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test.sh b/test.sh index 979b376c..c34bbaf6 100755 --- a/test.sh +++ b/test.sh @@ -5,7 +5,7 @@ OUTPUT_FILE=$1 touch $OUTPUT_FILE echo "| Name | Data Type | Input Shapes | torch2trt kwargs | Max Error | PSNR | MSE | Throughput (PyTorch) | Throughput (TensorRT) | Latency (PyTorch) | Latency (TensorRT) |" >> $OUTPUT_FILE -echo "|------|-----------|--------------|------------------|-----------|----------------------|-----------------------|-------------------|--------------------|" >> $OUTPUT_FILE +echo "|------|-----------|--------------|------------------|-----------|------|-----|----------------------|-----------------------|-------------------|--------------------|" >> $OUTPUT_FILE python3 -m torch2trt.test -o $OUTPUT_FILE --name alexnet --include=torch2trt.tests.torchvision.classification python3 -m torch2trt.test -o $OUTPUT_FILE --name squeezenet1_0 --include=torch2trt.tests.torchvision.classification