Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-09 19:02:25 +02:00
parent 8a575f6447
commit a8fd4cabf3
28 changed files with 36 additions and 32 deletions
+3 -3
View File
@@ -32,13 +32,13 @@ function print_output {
out_file=results.log
rm $out_file
modes=( 1 ) # only FP32
# modes=( 1 2 ) # FP32 and FP16
# modes=( 1 ) # only FP32
modes=( 1 2 ) # FP32 and FP16
# modes=( 1 2 3 ) # FP32, FP16 and INT8
for i in "${modes[@]}"
do
rm *rt
# rm *rt
if [ $i -eq 1 ]
then
export TKDNN_MODE=FP32