minor refactoring

maet3608 · Sep 13, 2017 · af95270 · af95270
1 parent f991d18
commit af95270
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nutsml/network.py b/nutsml/network.py
@@ -279,12 +279,12 @@ def load_weights(self, weightspath=None):
             param.set_value(weights[name])
 
     def print_layers(self):
-        import lasagne as nn
+        import lasagne as la
         layers = list(LasagneNetwork._layers(self.out_layer, ret_input=True))
         for i, layer in enumerate(reversed(layers)):
             print('_' * 80)
             name = layer.__class__.__name__
-            shape = nn.layers.get_output_shape(layer)
+            shape = la.layers.get_output_shape(layer)
             print('{:3d}  {:30s} {}'.format(i, name, shape), end=' ')
             if hasattr(layer, 'filter_size'):
                 print('{}'.format(layer.filter_size[0]), end='//')