From ec3481477d5fdfcdbab387dd77519101d45f1c8f Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Fri, 23 Apr 2021 06:54:51 +0000
Subject: [PATCH] fix layer tools

---
 deepspeech/utils/layer_tools.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/deepspeech/utils/layer_tools.py b/deepspeech/utils/layer_tools.py
index 1c3105605..429391874 100644
--- a/deepspeech/utils/layer_tools.py
+++ b/deepspeech/utils/layer_tools.py
@@ -56,8 +56,8 @@ def print_params(model, print_func=print):
         if print_func:
             print_func(msg)
     if print_func:
-        total = total / 1024**3
-        print_func(f"Total parameters: {num_params}, {total}G elements.")
+        total = total / 1024**2
+        print_func(f"Total parameters: {num_params}, {total:.4f}M elements.")
 
 
 def gradient_norm(layer: nn.Layer):