Commit 3b8bc043 authored by peastman's avatar peastman
Browse files

Bug fix

parent d471172e
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -138,7 +138,7 @@ class SeqToSeq(TensorGraph):
      self._embedding_stddev = layers.Dense(
          embedding_dimension, in_layers=prev_layer)
      prev_layer = layers.CombineMeanStd(
          [self._embedding_mean, self._embedding_stddev])
          [self._embedding_mean, self._embedding_stddev], training_only=True)
    return prev_layer

  def _create_decoder(self, n_layers, dropout, embedding_dimension):