[Benchmarking Py] Removing Max Workspace Size Args from Base Scripts

Author: DEKHTIARJonathan

tftrt/benchmarking-python/huggingface/bert/base_run_inference.sh

@@ -60,7 +60,6 @@ BYPASS_ARGUMENTS=$(echo ${BYPASS_ARGUMENTS} | tr -s " ")
 
 MIN_SEGMENT_SIZE=5
 VOCAB_SIZE=-1
-MAX_WORKSPACE_SIZE=$((2 ** (32 + 1)))  # + 1 necessary compared to python
 MAX_SAMPLES=1
 OUTPUT_TENSORS_NAME="last_hidden_state,pooler_output"
 
@@ -93,7 +92,6 @@ echo ""
 # Custom Transormer Task Flags
 echo "[*] VOCAB_SIZE: ${VOCAB_SIZE}"
 echo "[*] SEQ_LEN: ${SEQ_LEN}"
-echo "[*] MAX_WORKSPACE_SIZE: ${MAX_WORKSPACE_SIZE}"
 echo "[*] MAX_SAMPLES: ${MAX_SAMPLES}"
 echo "[*] OUTPUT_TENSORS_NAME: ${OUTPUT_TENSORS_NAME}"
 echo ""
@@ -148,7 +146,6 @@ python ${BASE_DIR}/infer.py \
     --batch_size ${BATCH_SIZE} \
     --vocab_size ${VOCAB_SIZE} \
     --sequence_length=${SEQ_LEN} \
-    --max_workspace_size ${MAX_WORKSPACE_SIZE} \
     --total_max_samples=${MAX_SAMPLES} \
     --output_tensors_name=${OUTPUT_TENSORS_NAME} \
     ${BYPASS_ARGUMENTS}

tftrt/benchmarking-python/object_detection/base_run_inference.sh

@@ -51,16 +51,9 @@ BYPASS_ARGUMENTS=$(echo ${BYPASS_ARGUMENTS} | tr -s " ")
 # ============== Set model specific parameters ============= #
 
 INPUT_SIZE=640
-MAX_WORKSPACE_SIZE=$((2 ** (32 + 1)))  # + 1 necessary compared to python
 MAX_SAMPLES=5000
 OUTPUT_TENSORS_NAME="boxes,classes,num_detections,scores"
 
-case ${MODEL_NAME} in
-  "faster_rcnn_resnet50_coco" | "ssd_mobilenet_v1_fpn_coco")
-    MAX_WORKSPACE_SIZE=$((2 ** (24 + 1)))  # + 1 necessary compared to python
-    ;;
-esac
-
 # %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% #
 
 echo -e "\n********************************************************************"
@@ -72,7 +65,6 @@ echo ""
 # Custom Object Detection Task Flags
 echo "[*] BATCH_SIZE: ${BATCH_SIZE}"
 echo "[*] INPUT_SIZE: ${INPUT_SIZE}"
-echo "[*] MAX_WORKSPACE_SIZE: ${MAX_WORKSPACE_SIZE}"
 echo "[*] MAX_SAMPLES: ${MAX_SAMPLES}"
 echo "[*] OUTPUT_TENSORS_NAME: ${OUTPUT_TENSORS_NAME}"
 echo ""
@@ -151,7 +143,6 @@ python ${BASE_DIR}/infer.py \
     --model_source "tf_models_object" \
     --batch_size ${BATCH_SIZE} \
     --input_size ${INPUT_SIZE} \
-    --max_workspace_size ${MAX_WORKSPACE_SIZE} \
     --total_max_samples=${MAX_SAMPLES} \
     --output_tensors_name=${OUTPUT_TENSORS_NAME} \
     ${BYPASS_ARGUMENTS}