Minor Changes

Author: DEKHTIARJonathan

tftrt/examples/benchmark_runner.py

@@ -224,12 +224,18 @@ def engine_build_input_fn(num_batches, model_phase):
                         f"`{self._args.output_saved_model_dir}`"
                     )
 
-        savedmodel_outputs = graph_func.structured_outputs
+        savedmodel_outputs = print_dict(
+            graph_func.structured_outputs,
+            redirect_to_str=True
+        )
 
-        chosen_outputs = sorted(self._args.output_tensors_name.split(","))
+        chosen_outputs = "\n  - ".join(
+            sorted(self._args.output_tensors_name.split(","))
+        )
 
-        self._debug_print(f"Available Output Tensors: {savedmodel_outputs}")
-        self._debug_print(f"Chosen Output Tensor: {chosen_outputs}")
+        self._debug_print(f"Available Output Tensors:\n{savedmodel_outputs}")
+        print()  # visual spacing
+        self._debug_print(f"Chosen Output Tensor:\n  - {chosen_outputs}")
 
         return graph_func
 

tftrt/examples/benchmark_utils.py

@@ -23,16 +23,18 @@ def wrapper(*args, **kwargs):
     return wrapper
 
 
-def print_dict(input_dict, prefix='  ', postfix=''):
-    for k, v in sorted(input_dict.items()):
-        print(
-            '{prefix}{arg_name}: {value}{postfix}'.format(
-                prefix=prefix,
-                arg_name=k,
-                value='%.1f' % v if isinstance(v, float) else v,
-                postfix=postfix
-            )
-        )
+def print_dict(input_dict, prefix='  ', postfix='', redirect_to_str=False):
+    rslt_str = ""
+    for key, val in sorted(input_dict.items()):
+        val = f"{val:.1f}" if isinstance(val, float) else val
+        tmp_str = f"{prefix}- {key}: {val}{postfix}"
+        if not redirect_to_str:
+            print(tmp_str)
+        else:
+            rslt_str += f"{tmp_str}\n"
+
+    if redirect_to_str:
+        return rslt_str.rstrip()
 
 
 @contextmanager

tftrt/examples/image_classification/scripts/base_script.sh

@@ -163,11 +163,12 @@ COMMAND="${PREPEND_COMMAND} python image_classification.py \
     --input_size ${INPUT_SIZE} \
     --preprocess_method ${PREPROCESS_METHOD} \
     --num_classes ${NUM_CLASSES} \
+    --total_max_samples=49920 \
     ${OUTPUT_TENSOR_IDX_FLAG} \
     ${OUTPUT_TENSOR_NAME_FLAG} \
     ${BYPASS_ARGUMENTS}"
 
-COMMAND=$(echo "${COMMAND}" | tr -s " ")
+COMMAND=$(echo ${COMMAND} | sed 's/ *$//g')  # Trimming whitespaces
 
 echo -e "**Executing:**\n\n${COMMAND}\n"
 sleep 5

tftrt/examples/object_detection/scripts/base_script.sh

@@ -152,7 +152,7 @@ COMMAND="${PREPEND_COMMAND} python object_detection.py \
     --max_workspace_size ${MAX_WORKSPACE_SIZE} \
     ${BYPASS_ARGUMENTS}"
 
-COMMAND=$(echo "${COMMAND}" | tr -s " ")
+COMMAND=$(echo ${COMMAND} | sed 's/ *$//g')  # Trimming whitespaces
 
 echo -e "**Executing:**\n\n${COMMAND}\n"
 sleep 5

tftrt/examples/transformers/scripts/base_script.sh

@@ -138,7 +138,7 @@ COMMAND="${PREPEND_COMMAND} python transformers.py \
     --minimum_segment_size ${MIN_SEGMENT_SIZE} \
     ${BYPASS_ARGUMENTS}"
 
-COMMAND=$(echo "${COMMAND}" | tr -s " ")
+COMMAND=$(echo ${COMMAND} | sed 's/ *$//g')  # Trimming whitespaces
 
 echo -e "**Executing:**\n\n${COMMAND}\n"
 sleep 5