Fixes

Signed-off-by: Joaquin Anton Guirao <[email protected]>

internal_tools/hw_decoder_bench.py

@@ -394,7 +394,7 @@ def vit_pipeline(is_training=False, image_shape=(384, 384, 3), num_classes=1000)
     iter_duration = iter_end_time - iter_start_time
     execution_times.append(iter_duration)
 
-    if args.print_every_n_iterations > 0 and iteration % args.print_every_n_iterations == 0:
+    if args.print_every_n_iterations > 0 and (iteration % args.print_every_n_iterations == 0 or iteration == 0 or iteration == test_iterations - 1):
         elapsed_time = time.perf_counter() - start_time
         throughput = (iteration + 1) * args.batch_size * args.gpu_num / elapsed_time
         mean_t = statistics.mean(execution_times)