Skip to content

D
DeepSpeech

PaddlePaddle / DeepSpeech
接近 2 年 前同步成功

通知 210
Star 8425
Fork 1598
D
DeepSpeech
提交 ec348147 编写于 作者: H Hui Zhang
浏览文件
操作

fix layer tools

上级 b5339633
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
deepspeech/utils/layer_tools.py
浏览文件 @ ec348147
...@@ -56,8 +56,8 @@ def print_params(model, print_func=print): ...@@ -56,8 +56,8 @@ def print_params(model, print_func=print):
if print_func: if print_func:
print_func(msg) print_func(msg)
if print_func: if print_func:
total = total / 1024**3 total = total / 1024**2
print_func(f"Total parameters: {num_params}, {total}G elements.") print_func(f"Total parameters: {num_params}, {total:.4f}M elements.")
def gradient_norm(layer: nn.Layer): def gradient_norm(layer: nn.Layer):
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或