Commit 04e104b7 authored by leswing's avatar leswing
Browse files

add back in batch norm

parent 8be6317b

deepchem/models/tensorgraph/models/seqtoseq.py

+1 −1
Original line number Diff line number Diff line
@@ -514,7 +514,7 @@ class AspuruGuzikAutoEncoder(SeqToSeq): 
    prev_layer = layers.Flatten(prev_layer)

    prev_layer = layers.Dense(

        self._decoder_dimension, in_layers=prev_layer, activation_fn=tf.nn.relu)

    #prev_layer = layers.BatchNorm(prev_layer)

    prev_layer = layers.BatchNorm(prev_layer)

    if self._variational:

      self._embedding_mean = layers.Dense(

          self._embedding_dimension,

CRA Git | Maintained and supported by SUSTech CRA and CCSE