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()