add run_benchmark.sh

Coderx7 · Coderx7 · commit 94c0f032eeef · 2023-02-14T21:35:18.000+03:30
diff --git a/ImageNet/training_scripts/imagenet_training/run_benchmark.sh b/ImageNet/training_scripts/imagenet_training/run_benchmark.sh
@@ -0,0 +1,10 @@
+echo 'benchmarking list_small.txt'
+python benchmark.py --model-list model_list_small.txt --bench inference --channels-last --results-file results/benchmark_inference_RTX3080_fp32_small.csv
+
+echo 'benchmarking list_normal_known'
+python benchmark.py --model-list model_list_normal_known.txt --bench inference --channels-last --results-file results/benchmark_inference_RTX3080_fp32_normal_known.csv
+
+echo 'benchmarking list_normal'
+python benchmark.py --model-list model_list_normal.txt --bench inference --channels-last --results-file results/benchmark_inference_RTX3080_fp32_normal.csv
+
+echo 'all done'
