fix layer tools

4 years ago · ec3481477d
parent b5339633e3
commit ec3481477d
1 changed files with 2 additions and 2 deletions
--- 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):