From 6754999c3977a1f16e8f4c6048d345e40a948473 Mon Sep 17 00:00:00 2001
From: pinto0309 <rmsdh122@yahoo.co.jp>
Date: Wed, 20 Sep 2023 21:15:09 +0900
Subject: [PATCH] thousands separator
---
ssc4onnx/__init__.py | 2 +-
ssc4onnx/onnx_structure_check.py | 6 +++---
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/ssc4onnx/__init__.py b/ssc4onnx/__init__.py
index 3f11bfb..bf08bab 100644
--- a/ssc4onnx/__init__.py
+++ b/ssc4onnx/__init__.py
@@ -1,3 +1,3 @@
from ssc4onnx.onnx_structure_check import structure_check, main
-__version__ = '1.0.5'
+__version__ = '1.0.6'
diff --git a/ssc4onnx/onnx_structure_check.py b/ssc4onnx/onnx_structure_check.py
index 71f88c5..a4db2f9 100644
--- a/ssc4onnx/onnx_structure_check.py
+++ b/ssc4onnx/onnx_structure_check.py
@@ -57,7 +57,7 @@ def human_readable_size(num, suffix="B"):
if abs(num) < 1024.0:
return f"{num:3.1f}{unit}{suffix}"
num /= 1024.0
- return f"{num:.1f}Yi{suffix}"
+ return f"{num:,.1f}Yi{suffix}"
class ModelInfo:
@@ -147,10 +147,10 @@ def structure_check(
table.add_column('OP Type')
table.add_column('OPs')
sorted_list = sorted(list(set(model_info.op_nums.keys())))
- _ = [table.add_row(key, str(model_info.op_nums[key])) for key in sorted_list]
+ _ = [table.add_row(key, f"{model_info.op_nums[key]:,}") for key in sorted_list]
table.add_row('----------------------', '----------')
ops_count = sum([model_info.op_nums[key] for key in sorted_list])
- table.add_row('Total number of OPs', str(ops_count))
+ table.add_row('Total number of OPs', f"{ops_count:,}")
table.add_row('======================', '==========')
table.add_row('Model Size', human_readable_size(model_info.model_size))
rich_print(table)