Prevent creation of loss on empty network.

lmjohns3 · Aug 31, 2015 · 2fd28be · 2fd28be
1 parent db1240e
commit 2fd28be
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/theanets/graph.py b/theanets/graph.py
@@ -116,10 +116,11 @@ def __init__(self, layers, loss='mse', weighted=False):
         logging.info('network has %d total parameters', self.num_params)
 
         self.losses = []
-        self.set_loss(loss,
-                      weighted=weighted,
-                      target=self.INPUT_NDIM,
-                      output_name=self.layers[-1].output_name())
+        if loss and self.layers:
+            self.set_loss(loss,
+                          weighted=weighted,
+                          target=self.INPUT_NDIM,
+                          output_name=self.layers[-1].output_name())
 
     def add_layer(self, layer=None, is_output=False, **kwargs):
         '''Add a layer to our network graph.