testadmin
/
Parakeet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix typos

This commit is contained in:
iclementine 2020-11-19 22:20:31 +08:00
parent 2dce0887b3
commit fb49c1e77d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
parakeet/utils/layer_tools.py
View File

@ -26,7 +26,7 @@ def summary(layer: nn.Layer):
print("layer has {} parameters, {} elements.".format(num_params, print("layer has {} parameters, {} elements.".format(num_params,
num_elements)) num_elements))
def gradien_norm(layer: nn.Layer): def gradient_norm(layer: nn.Layer):
grad_norm_dict = {} grad_norm_dict = {}
for name, param in layer.state_dict().items(): for name, param in layer.state_dict().items():
if param.trainable: if param.trainable: