Sync from tflite-micro at b671143.

Signed-off-by: CFU-Playground-Bot <cfu-playground-bot@google.com>
google · Aug 31, 2022 · 5a1e227 · 5a1e227
1 parent 6b0eeec
commit 5a1e227
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/conf/tflite-micro.version b/conf/tflite-micro.version
@@ -1 +1 @@
-3136969
+b671143
diff --git a/.../tflite-micro/tensorflow/lite/micro/examples/hello_world/train/train_hello_world_model.py b/.../tflite-micro/tensorflow/lite/micro/examples/hello_world/train/train_hello_world_model.py
@@ -58,6 +58,7 @@
 import textwrap
 import traceback
 from typing import NamedTuple, Optional
+from pathlib import Path
 
 ## Shell command utility method
 
@@ -606,7 +607,10 @@ def compare_sizes(config: ConfigData) -> None:
   """
 
   # Calculate size
-  size_tf = os.path.getsize(config.MODEL_TF)
+  size_tf = sum(
+      map(lambda p: p.stat().st_size,
+          filter(Path.is_file,
+                 Path(config.MODEL_TF).rglob('*'))))
   size_no_quant_tflite = os.path.getsize(config.MODEL_NO_QUANT_TFLITE)
   size_quant_tflite = os.path.getsize(config.MODEL_QUANT_TFLITE)