From 7f5c28d2292f6bc7799452e98f72eae2a42bb3b2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=AE=9D=E5=BC=9F?= Date: Tue, 12 Apr 2022 07:05:53 +0000 Subject: [PATCH 1/2] update --- .../UNet2D_ID0008_for_TensorFlow/utils/parse_results.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TensorFlow/built-in/cv/image_segmentation/UNet2D_ID0008_for_TensorFlow/utils/parse_results.py b/TensorFlow/built-in/cv/image_segmentation/UNet2D_ID0008_for_TensorFlow/utils/parse_results.py index 34ec762f4..a027d5caa 100644 --- a/TensorFlow/built-in/cv/image_segmentation/UNet2D_ID0008_for_TensorFlow/utils/parse_results.py +++ b/TensorFlow/built-in/cv/image_segmentation/UNet2D_ID0008_for_TensorFlow/utils/parse_results.py @@ -22,7 +22,7 @@ def process_performance_stats(timestamps, batch_size): latency_ms = timestamps_ms.mean() std = timestamps_ms.std() n = np.sqrt(len(timestamps_ms)) - throughput_imgps = (1000.0 * batch_size / timestamps_ms).mean() + throughput_imgps = (10 * 1000.0 * batch_size / timestamps_ms).mean() stats = [("Throughput Avg", str(throughput_imgps)), ('Latency Avg:', str(latency_ms))] -- Gitee From 3a6fedb733c3b34196cbcc9b9e4368e79b1d9000 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=AE=9D=E5=BC=9F?= Date: Tue, 12 Apr 2022 07:06:28 +0000 Subject: [PATCH 2/2] update --- .../UNet2D_ID0008_for_TensorFlow/utils/setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TensorFlow/built-in/cv/image_segmentation/UNet2D_ID0008_for_TensorFlow/utils/setup.py b/TensorFlow/built-in/cv/image_segmentation/UNet2D_ID0008_for_TensorFlow/utils/setup.py index 753d05c60..8292db7d5 100644 --- a/TensorFlow/built-in/cv/image_segmentation/UNet2D_ID0008_for_TensorFlow/utils/setup.py +++ b/TensorFlow/built-in/cv/image_segmentation/UNet2D_ID0008_for_TensorFlow/utils/setup.py @@ -87,7 +87,7 @@ def build_estimator(params, model_dir): #save_checkpoints_steps=(params.max_steps // hvd.size()) if hvd.rank() == 0 else None, save_checkpoints_steps=params.max_steps, precision_mode='allow_fp32_to_fp16', - iterations_per_loop=1, + iterations_per_loop=10, hcom_parallel=True, keep_checkpoint_max=1) # for npu -- Gitee