From 7a868c51ae29c7c85b22fac05a731a43ab51e44e Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 16:10:32 -0400 Subject: [PATCH] removed dead code in grads --- pytorch_lightning/root_module/grads.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/pytorch_lightning/root_module/grads.py b/pytorch_lightning/root_module/grads.py index e4d1701b..8ed17a3e 100644 --- a/pytorch_lightning/root_module/grads.py +++ b/pytorch_lightning/root_module/grads.py @@ -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)) \ No newline at end of file