edit.py

      old = lasagne.layers.get_all_param_values(self.output_layer)
      new = []
      for layer in old:
        shape = layer.shape
        if len(shape)<2:
          shape = (shape[0], 1)
        W= lasagne.init.GlorotUniform()(shape)
        if W.shape != layer.shape:
          W = np.squeeze(W, axis= 1)
        new.append(W)
      lasagne.layers.set_all_param_values(self.output_layer, new)