From 97aa69c8f633245e09c8443fafaf447a215c3330 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 16:16:26 -0400 Subject: [PATCH] removed dead code in grads --- pytorch_lightning/root_module/memory.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/root_module/memory.py b/pytorch_lightning/root_module/memory.py index 17f20efe..0bafe562 100644 --- a/pytorch_lightning/root_module/memory.py +++ b/pytorch_lightning/root_module/memory.py @@ -115,11 +115,13 @@ class ModelSummary(object): ''' df = pd.DataFrame( np.zeros( (len(self.layer_names), 3) ) ) - df.columns = ['Name', 'Type', 'Params'] + df.columns = ['Name', 'Type', 'Params', 'In_sizes', 'Out_sizes'] df['Name'] = self.layer_names df['Type'] = self.layer_types df['Params'] = self.param_nums + df['In_sizes'] = self.in_sizes + df['Out_sizes'] = self.out_sizes self.summary = df return @@ -128,6 +130,7 @@ class ModelSummary(object): self.get_layer_names() self.get_parameter_sizes() self.get_parameter_nums() + self.get_variable_sizes() self.make_summary()