Commit 6cef6f4d authored by leswing's avatar leswing
Browse files

Axis=0 for concat

parent 0ffdcfe6
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -102,7 +102,7 @@ class WeaveTensorGraph(TensorGraph):
        cost = L2Loss(in_layers=[label, regression])
        costs.append(cost)

    all_cost = Concat(in_layers=costs)
    all_cost = Concat(in_layers=costs, axis=0)
    self.weights = Weights(shape=(None, self.n_tasks))
    loss = WeightedError(in_layers=[all_cost, self.weights])
    self.set_loss(loss)
+1 −0
Original line number Diff line number Diff line
@@ -327,6 +327,7 @@ class TensorGraph(Model):
      for node in order:
        with tf.name_scope(node):
          node_layer = self.layers[node]
          print(node)
          node_layer.create_tensor(training=self._training_placeholder)
      self.built = True