wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

removed dead code in grads

Browse Source
This commit is contained in:
William Falcon
2019-07-24 16:10:32 -04:00
parent 2f4bd676e8
commit 7a868c51ae
1 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/grads.py
-11
View File
@@ -27,14 +27,3 @@ class GradInformation(nn.Module):
results['grad_{}_norm_total'.format(norm_type)] = round(total_norm.data.cpu().numpy().flatten()[0], 3)
return results
def describe_grads(self):
for p in self.parameters():
g = p.grad.data.numpy().flatten()
print(np.max(g), np.min(g), np.mean(g))
def describe_params(self):
for p in self.parameters():
g = p.data.numpy().flatten()
print(np.max(g), np.min(g), np.mean(g))